code: def setUp(self):
      super().setUp()
      self.key = jax.random.PRNGKey(0)
      self.action_shape = (3,)
      self.num_layers = 3
  
      self.example_action = jnp.zeros(
          self.action_shape, dtype=jnp.float32
      )
      self.example_state = jnp.zeros((8,), dtype=jnp.float32)
  
      self.network_def = continuous_networks.SACNetwork(
          self.action_shape, self.num_layers
      )
      self.params = self.network_def.init(
          self.key, self.example_state, self.key
      )
  
file_path: dopamine/tests/dopamine/jax/continuous_networks_test.py
imports: from absl.testing import absltest
  from dopamine.jax import continuous_networks
  import jax
  from jax import numpy
label_desc: Use image node_id 1 for calling the SacNetworkTest obj's underlying member method code with example usage: obj.setUp() without return types