code: def test_set_get_architecture(self, architecture):
      # Force graph mode
      with tf.compat.v1.Graph().as_default():
          directory = self.get_temp_dir()
          with self.test_session(graph=tf.Graph()) as sess:
              architecture.save_to_graph()
              saver = tf.compat.v1.train.Saver()
              sess.run(tf.compat.v1.global_variables_initializer())
              sess.run(tf.compat.v1.local_variables_initializer())
              saver.save(sess, directory + "/ckpt")
  
          output_architecture = (
              graph_architecture.restore_from_checkpoint(directory)
          )
          self.assertAllEqual(output_architecture, architecture)
  
file_path: model_search/model_search/architecture/graph_architecture_test.py
imports: from absl.testing import parameterized
  from model_search.architecture import architecture_utils
  from model_search.architecture import graph_architecture
  from model_search.proto import phoenix_spec_pb2
  import numpy
  import tensorflow.compat.v2
label_desc: Use image node_id 1 for calling the GraphArchitectureTest obj's underlying member method code with example usage: obj.test_set_get_architecture(architecture) without return types