class_docstr: MockTFEnvironment.
  
  Stores all actions taken in `actions_taken`. The returned values are:
  
  step: FIRST, 1., 0., [0]
  step: MID, 1., 0., [1]
  step: LAST, 0., 1. [2]
  ...repeated
class_docstr_tok: ["MockTFEnvironment",".","Stores","all","actions","taken","in","`","actions_taken","`",".","The","returned","values","are",":","step",":","FIRST",",","1.",",","0.",",","[","0","]","step",":","MID",",","1.",",","0.",",","[","1","]","step",":","LAST",",","0.",",","1",".","[","2","]","...","repeated"]
code: def _reset(self):
      increase_resets = self.resets.assign_add(1)
      with tf.control_dependencies([increase_resets]):
          reset_op = self._state.assign(self._initial_state)
      with tf.control_dependencies([reset_op]):
          time_step = self.current_time_step()
      return time_step
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  from tf_agents import specs
  from tf_agents.environments import tf_environment
  from tf_agents.trajectories import time_step
  from tf_agents.utils import common
label_desc: Use image node_id 3 for calling the TFEnvironmentMock obj's underlying member method code with example usage: obj._reset() and returns: time_step