code: def test_network_builds(self):
      env = suite_gym.load("CartPole-v0")
      tf_env = tf_py_environment.TFPyEnvironment(env)
      rnn_network = q_rnn_network.QRnnNetwork(
          tf_env.observation_spec(),
          tf_env.action_spec(),
          lstm_size=(40,),
      )
  
      first_time_step = tf_env.current_time_step()
      q_values, state = rnn_network(
          first_time_step.observation,
          first_time_step.step_type,
          network_state=rnn_network.get_initial_state(batch_size=1),
      )
      self.assertEqual((1, 2), q_values.shape)
      self.assertEqual((1, 40), state[0].shape)
      self.assertEqual((1, 40), state[1].shape)
  
file_path: agents/tf_agents/networks/q_rnn_network_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  from tf_agents.environments import suite_gym
  from tf_agents.environments import tf_py_environment
  from tf_agents.networks import expand_dims_layer
  from tf_agents.networks import q_rnn_network
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import time_step
label_desc: Use image node_id 1 for calling the QRnnNetworkTest obj's underlying member method code with example usage: obj.test_network_builds() without return types