code: def test_save_load_model_architecture(self):
      config = {"lr": 0.2}
      import uuid
  
      model_path = os.path.join(
          tempfile.gettempdir(), str(uuid.uuid1()) + ".json"
      )
      try:
          model = simple_model(config)
          with open(model_path, "w") as f:
              f.write(model.to_json())
  
          from bigdl.dllib.utils.file_utils import enable_hdfs_load
  
          @enable_hdfs_load
          def load_model_architecture(path):
              with open(path, "rb") as f:
                  model = tf.keras.models.model_from_json(f.read())
              return model
  
          model_load = load_model_architecture(model_path)
          assert model.summary() == model_load.summary()
      finally:
          if os.path.exists(model_path):
              os.remove(model_path)
  
imports: import os
  import pytest
  import shutil
  import tempfile
  from unittest import TestCase
  import numpy
  import tensorflow
  from bigdl.orca.learn.tf2 import Estimator
  from bigdl.orca import OrcaContext
label_desc: Use image node_id 6 for calling the TestTF2Estimator obj's underlying member method code with example usage: obj.test_save_load_model_architecture() and returns: model