code: def test_registration_and_interaction_with_env_problem(self):
      batch_size = 5
      # This ensures that registration has occurred.
      ep = registry.env_problem(
          "reacher_env_problem", batch_size=batch_size
      )
      ep.reset()
      num_done = 0
      nsteps = 100
      for _ in range(nsteps):
          actions = np.stack(
              [ep.action_space.sample() for _ in range(batch_size)]
          )
          obs, rewards, dones, infos = ep.step(actions)
  
          # Assert that things are happening batchwise.
          self.assertEqual(batch_size, len(obs))
          self.assertEqual(batch_size, len(rewards))
          self.assertEqual(batch_size, len(dones))
          self.assertEqual(batch_size, len(infos))
  
          done_indices = env_problem_utils.done_indices(dones)
          ep.reset(done_indices)
          num_done += sum(dones)
  
      # Assert that something got done atleast,
      self.assertGreater(num_done, 0)
  
file_path: tensor2tensor/tensor2tensor/envs/mujoco_problems_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  from tensor2tensor.envs import env_problem_utils
  from tensor2tensor.envs import mujoco_problems
  from tensor2tensor.utils import registry
  import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling the ReacherEnvProblemTest obj's underlying member method code with example usage: obj.test_registration_and_interaction_with_env_problem() without return types