class_docstr: Base wrapper forwarding the DM control types into the tf_agents ones.
class_docstr_tok: ["Base","wrapper","forwarding","the","DM","control","types","into","the","tf_agents","ones","."]
code: def _step(self, action):
      action = tf.nest.map_structure(
          lambda a, s: np.asarray(a, dtype=s.dtype),
          action,
          self._env.action_spec(),
      )
      return convert_time_step(self._env.step(action))
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import functools
  import numpy
  import tensorflow
  from tf_agents.environments import wrappers
  from tf_agents.specs import array_spec
  from tf_agents.trajectories import time_step
label_desc: Use image node_id 4 for calling the DmControlWrapper obj's underlying member method code with example usage: obj._step(action) and returns: convert_time_step