code: def test_does_not_update_freeze_module():
      state_dict = {
          "module_dict.sentence__ludwig.encoder_obj.transformer.module.embeddings.LayerNorm.bias": 0.0,
          "sentence__ludwig.encoder_obj.transformer.module.encoder.layer.0.attention.output.LayerNorm.weight": 0.0,
          "module_dict.sentence__ludwig.encoder_obj.transformer.module.embeddings.word_embeddings.weight": 0.0,
      }
  
      # Ensures that models saved with FreezeModule aren't modified.
      updated_state_dict = update_state_dict(state_dict)
      assert updated_state_dict == state_dict
  
label_desc: Use image node_id 2 for calling a global function with example usage: test_does_not_update_freeze_module() without return types