code: def setup_method(self):
      super().setup_method()
      self.batch_size = 1
      self.time_step = 1
      self.feature_shape = 2
      self.output_dim = 1
  
      self.obs_inputs = np.full(
          (self.batch_size, self.time_step, self.feature_shape), 1.0
      )
      self.obs_input = np.full(
          (self.batch_size, self.feature_shape), 1.0
      )
  
      self._input_var = tf.compat.v1.placeholder(
          tf.float32,
          shape=(None, None, self.feature_shape),
          name="input",
      )
      self._step_input_var = tf.compat.v1.placeholder(
          tf.float32, shape=(None, self.feature_shape), name="input"
      )
  
file_path: garage/tests/garage/tf/models/test_categorical_lstm_model.py
imports: import pickle
  import numpy
  import pytest
  import tensorflow
  import tensorflow_probability
  from garage.tf.models import CategoricalLSTMModel
  from tests.fixtures import TfGraphTestCase
label_desc: Use image node_id 1 for calling the TestCategoricalLSTMModel obj's underlying member method code with example usage: obj.setup_method() without return types