code: def setUp(self):
      super().setUp()
      self.key = jax.random.PRNGKey(0)
      self.input_shape = (84, 84, 3)
      self.example_input = jnp.full(
          self.input_shape, 128, dtype=jnp.uint8
      )
  
file_path: dopamine/tests/dopamine/labs/sac_from_pixels/continuous_networks_test.py
imports: from absl.testing import absltest
  from dopamine.labs.sac_from_pixels import continuous_networks
  import jax
  from jax import numpy
label_desc: Use image node_id 1 for calling the SACEncoderNetworkTest obj's underlying member method code with example usage: obj.setUp() without return types