code: def __init__(self, action_spec: specs.DiscreteArray):
      super().__init__(name="r2d2_test_network")
      self._net = snt.DeepRNN(
          [
              snt.Flatten(),
              snt.LSTM(20),
              snt.nets.MLP([50, 50, action_spec.num_values]),
          ]
      )
  
file_path: acme/acme/agents/tf/r2d3/agent_test.py
imports: import acme
  from acme import specs
  from acme.agents.tf import r2d3
  from acme.agents.tf.dqfd import bsuite_demonstrations
  from acme.testing import fakes
  from acme.tf import networks
  import dm_env
  import numpy
  import sonnet
  from absl.testing import absltest
label_desc: Use image node_id 1 to create a new SimpleNetwork object from inherited base classes: networks with example: obj = SimpleNetwork(action_spec)