code: def _assert_changes_post_prepare(
      original_module,
      palettized_module,
      n_bits,
      cluster_dim,
      kmeans_max_iter,
  ):
      assert (
          type(palettized_module)
          == quantization_mappings.DEFAULT_QAT_MODULE_MAPPINGS[
              type(original_module)
          ]
      )
      assert (
          palettized_module.weight_fake_quant.n_clusters[0]
          == 2**n_bits
      )
      assert (
          palettized_module.weight_fake_quant.cluster_dim == cluster_dim
      )
      assert (
          palettized_module.weight_fake_quant.kmeans_max_iter
          == kmeans_max_iter
      )
  
label_desc: Use image node_id 2 for calling a global function with example usage: _assert_changes_post_prepare(original_module, palettized_module, n_bits, cluster_dim, kmeans_max_iter) without return types