code: def test_attach_config_simple_model_uniform_palettization_config(
      simple_model,
  ):
      config = DKMPalettizerConfig.from_dict(
          {"global_config": {"n_bits": 4}}
      )
      palettizer = DKMPalettizer(simple_model, config)
      palettizer.prepare(inplace=True)
  
      n_bits = config.global_config.n_bits
      _assert_changes_post_attach(simple_model.conv2, n_bits, 1)
      _assert_changes_post_attach(simple_model.fc1, n_bits, 1)
      _assert_changes_post_attach(simple_model.fc2, n_bits, 1)
  
imports: import copy
  import pytest
  import torch
  import torch.functional
  import torch.nn
  from coremltools.optimize.torch.palettization import DKMPalettizer, DKMPalettizerConfig
  from coremltools.optimize.torch.palettization.palettization_config import DEFAULT_PALETTIZATION_SCHEME
  from coremltools.test.optimize.torch.palettization.palettization_utils import _assert_changes_post_attach, _assert_changes_post_prepare
label_desc: Use image node_id 4 for calling a global function with example usage: test_attach_config_simple_model_uniform_palettization_config(simple_model) without return types