code: def test_reset_all(self):
      bt = trajectory.BatchTrajectory(batch_size=self.BATCH_SIZE)
  
      indices = np.arange(self.BATCH_SIZE)
      (
          observations,
          _,
          _,
          _,
      ) = self.get_random_observations_rewards_actions_dones()
  
      # Call reset.
      bt.reset(indices, observations)
  
      # Assert that all trajectories are active and not done (reset never marks
      # anything as done).
      self.assertTrue(all(t.is_active for t in bt.trajectories))
      self.assertEqual(0, bt.num_completed_trajectories)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import numpy
  from tensor2tensor.envs import time_step
  from tensor2tensor.envs import trajectory
  import tensorflow.compat.v1
  from tensorflow.compat.v1.io import gfile
label_desc: Use image node_id 3 for calling the BatchTrajectoryTest obj's underlying member method code with example usage: obj.test_reset_all() without return types