code: def setup_method(self):
      super().setup_method()
      batch_size = 5
      input_width = 10
      input_height = 10
      self._input_dim = (input_width, input_height, 3)
      self._obs_input = np.ones(
          (batch_size, 1, input_width, input_height, 3)
      )
      self._input_shape = (
          input_width,
          input_height,
          3,
      )  # height, width, channel
      self._input_ph = tf.compat.v1.placeholder(
          tf.float32,
          shape=(None, None) + self._input_shape,
          name="input",
      )
  
file_path: garage/tests/garage/tf/models/test_categorical_cnn_model.py
imports: import pickle
  import numpy
  import pytest
  import tensorflow
  import tensorflow_probability
  from garage.tf.models import CategoricalCNNModel
  from tests.fixtures import TfGraphTestCase
label_desc: Use image node_id 1 for calling the TestCategoricalMLPModel obj's underlying member method code with example usage: obj.setup_method() without return types