code: def test_trajectory_batch_stream_propagates_env_info(self):
      task = rl_task.RLTask(
          DummyEnv(), initial_trajectories=1, max_steps=4
      )
      stream = task.trajectory_batch_stream(
          batch_size=1, max_slice_length=4
      )
      tr_slice = next(stream)
      # control_mask = step % 2 == 0, discount_mask = step % 3 == 0.
      np.testing.assert_array_equal(
          tr_slice.env_info.control_mask, [[1, 0, 1, 0]]
      )
      np.testing.assert_array_equal(
          tr_slice.env_info.discount_mask, [[1, 0, 0, 1]]
      )
  
imports: import os
  from absl.testing import absltest
  import gym
  import numpy
  from trax import test_utils
  from trax.rl import task
label_desc: Use image node_id 26 for calling the TaskTest obj's underlying member method code with example usage: obj.test_trajectory_batch_stream_propagates_env_info() without return types