code: def setUp(self):
      super(TfxModelTest, self).setUp()
      self._path = tempfile.mkdtemp()
      input_layer = tf.keras.layers.Input(
          shape=(1), dtype=tf.string, name="example"
      )
      parsed_input = tf.io.parse_example(
          tf.reshape(input_layer, [-1]),
          {"input_0": tf.io.FixedLenFeature([1], dtype=tf.float32)},
      )
      output_layer = tf.keras.layers.Dense(1, name="output_0")(
          parsed_input["input_0"]
      )
      model = tf.keras.Model(input_layer, output_layer)
      model.compile(
          optimizer=tf.keras.optimizers.Adam(lr=0.001),
          loss=tf.keras.losses.binary_crossentropy,
      )
      model.save(self._path)
  
file_path: lit/lit_nlp/components/tfx_model_test.py
imports: import tempfile
  from lit_nlp.api import types
  from lit_nlp.components import tfx_model
  import tensorflow
label_desc: Use image node_id 1 for calling the TfxModelTest obj's underlying member method code with example usage: obj.setUp() without return types