code: def assert_critic_shapes_are_correct(
      self, critic_output: continuous_networks.SacCriticOutput
  ):
      with self.subTest(name="q_value1"):
          self.assertEqual(critic_output.q_value1.shape, (1,))
      with self.subTest(name="q_value2"):
          self.assertEqual(critic_output.q_value2.shape, (1,))
  
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 3 for calling the SacNetworkTest obj's underlying member method code with example usage: obj.assert_critic_shapes_are_correct(critic_output) without return types