code: def test_change_save_restore(self):
      model = ASRModel.from_pretrained("stt_en_conformer_ctc_small")
      model.change_attention_model(
          self_attention_model="rel_pos_local_attn",
          att_context_size=(64, 64),
      )
      attr_for_eq_check = [
          "encoder.self_attention_model",
          "encoder.att_context_size",
      ]
  
      with tempfile.TemporaryDirectory() as restore_folder:
          with tempfile.TemporaryDirectory() as save_folder:
              save_folder_path = save_folder
              # Where model will be saved
              model_save_path = os.path.join(
                  save_folder, f"{model.__class__.__name__}.nemo"
              )
              model.save_to(save_path=model_save_path)
              # Where model will be restored from
              model_restore_path = os.path.join(
                  restore_folder, f"{model.__class__.__name__}.nemo"
              )
              shutil.copy(model_save_path, model_restore_path)
          # at this point save_folder should not exist
          assert save_folder_path is not None and not os.path.exists(
              save_folder_path
          )
          assert not os.path.exists(model_save_path)
          assert os.path.exists(model_restore_path)
          # attempt to restore
          model_copy = model.__class__.restore_from(
              restore_path=model_restore_path,
              map_location=None,
              strict=True,
              return_config=False,
              override_config_path=None,
          )
  
          assert model.num_weights == model_copy.num_weights
          if (
              attr_for_eq_check is not None
              and len(attr_for_eq_check) > 0
          ):
              for attr in attr_for_eq_check:
                  assert getattr2(model, attr) == getattr2(
                      model_copy, attr
                  )
  
file_path: NeMo/tests/collections/asr/test_asr_local_attn.py
imports: import os
  import shutil
  import tempfile
  import pytest
  import pytorch_lightning
  import torch
  from omegaconf import DictConfig
  from nemo.collections.asr.models import ASRModel, EncDecCTCModel
label_desc: Use image node_id 2 for calling the TestASRLocalAttention obj's underlying member method code with example usage: obj.test_change_save_restore() without return types