parameters: (self)
return_statement: return rl_environment.TimeStep(
          observations=observations,
          rewards=None,
          discounts=None,
          step_type=rl_environment.StepType.FIRST)
docstring: Resets the environment.
docstring_summary: Resets the environment.
docstring_tokens: Resets the environment .
function: def reset(self):
      """Resets the environment."""
      self._should_reset = False
      self._time_counter = 0
      self._state = np.array([self._height - 1, 0])
  
      observations = {
          "info_state": [self._state.copy()],
          "legal_actions": [self._legal_actions],
          "current_player": 0,
      }
  
      return rl_environment.TimeStep(
          observations=observations,
          rewards=None,
          discounts=None,
          step_type=rl_environment.StepType.FIRST)
function_tokens: def reset ( self ) : self . _should_reset = False self . _time_counter = 0 self . _state = np . array ( [ self . _height - 1 , 0 ] ) observations = { "info_state" : [ self . _state . copy ( ) ] , "legal_actions" : [ self . _legal_actions ] , "current_player" : 0 , } return rl_environment . TimeStep ( observations = observations , rewards = None , discounts = None , step_type = rl_environment . StepType . FIRST )