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_utils.set_architecture(architecture)
              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 = architecture_utils.get_architecture(
              directory
          )
          self.assertAllEqual(output_architecture, architecture)
  
imports: from absl.testing import parameterized
  from model_search import block_builder
  from model_search import hparam
  from model_search.architecture import architecture_utils
  from model_search.metadata import trial
  from model_search.proto import phoenix_spec_pb2
  from model_search.proto import transfer_learning_spec_pb2
  import numpy
  import tensorflow.compat.v2
label_desc: Use image node_id 8 for calling the ArchitectureUtilsTest obj's underlying member method code with example usage: obj.test_set_get_architecture(architecture) without return types