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 __init__(
      self, initial_state=0, dtype=tf.int64, scope="TFEnviroment"
  ):
      self._dtype = dtype
      self._scope = scope
      self._initial_state = tf.cast(initial_state, dtype=self._dtype)
      observation_spec = specs.TensorSpec(
          [1], self._dtype, "observation"
      )
      action_spec = specs.BoundedTensorSpec(
          [], tf.int32, minimum=0, maximum=10
      )
      time_step_spec = ts.time_step_spec(observation_spec)
      super(TFEnvironmentMock, self).__init__(
          time_step_spec, action_spec
      )
      self._state = common.create_variable(
          "state", initial_state, dtype=self._dtype
      )
      self.steps = common.create_variable("steps", 0)
      self.episodes = common.create_variable("episodes", 0)
      self.resets = common.create_variable("resets", 0)
  
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 1 to create a new TFEnvironmentMock object from inherited base classes: tf_environment with example: obj = TFEnvironmentMock(initial_state, dtype, scope)