code: def test_model_is_pickleable(self):
      data = np.ones((3, 5))
      model = SimpleModel(output_dim=2)
  
      with tf.compat.v1.Session(graph=tf.Graph()) as sess:
          input_var = tf.compat.v1.placeholder(
              tf.float32, shape=(None, 5)
          )
          model.build(input_var)
  
          # assign bias to all one
          with tf.compat.v1.variable_scope(
              "SimpleModel/state", reuse=True
          ):
              bias = tf.compat.v1.get_variable("hidden_0/bias")
          bias.load(tf.ones_like(bias).eval())
  
          results = sess.run(
              model._networks["default"].outputs,
              feed_dict={model._networks["default"].input: data},
          )
          model_data = pickle.dumps(model)
  
      with tf.compat.v1.Session(graph=tf.Graph()) as sess:
          input_var = tf.compat.v1.placeholder(
              tf.float32, shape=(None, 5)
          )
          model_pickled = pickle.loads(model_data)
          outputs = model_pickled.build(input_var).outputs
  
          results2 = sess.run(outputs, feed_dict={input_var: data})
  
      assert np.array_equal(results, results2)
  
imports: import pickle
  import numpy
  import pytest
  import tensorflow
  from garage.tf.models.mlp import mlp
  from garage.tf.models.model import Model
  from tests.fixtures import TfGraphTestCase
label_desc: Use image node_id 8 for calling the TestModel obj's underlying member method code with example usage: obj.test_model_is_pickleable() without return types