class_docstr: Gym environment interface for Batched Environments (with batch size 1).
code: def step(self, action):
      obs, rewards, dones = self.batch_env.step([action])
      return obs[0], rewards[0], dones[0], {}
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from gym import Env
  import numpy
  from tensor2tensor.rl.envs.simulated_batch_env import SimulatedBatchEnv
  import tensorflow.compat.v1
label_desc: Use image node_id 2 for calling the FlatBatchEnv obj's underlying member method code with example usage: obj.step(action) and returns: obs,  rewards,  dones,  dict