original_docstring: 
          Steps the oracle's internal state forward with the agent's current action.
          :param agent_action: The action the agent plans to take.
          
docstring: Steps the oracle's internal state forward with the agent's current action.
docstring_tokens: Steps the oracle ' s internal state forward with the agent ' s current action .
code: def step(self, agent_action, oracle):
          """
          Steps the oracle's internal state forward with the agent's current action.
          :param agent_action: The action the agent plans to take.
          """
          self.last_action = self.next_action
          oracle, replan_output = self.replan(oracle, agent_action)
          self.next_action, next_subgoal = replan_output
          # If it's explore, mask out the object
          if np.argmax(next_subgoal[:4]) == 3:
              next_subgoal[-2:] = 0
          self.next_subgoal = next_subgoal
          # Don't include open subgoals
          if not np.argmax(next_subgoal[:4]) == 0:
              self.next_subgoal = next_subgoal
          self.last_step_error = False
          self.steps_since_lastfeedback += 1
          return oracle
code_tokens: def step ( self , agent_action , oracle ) : self . last_action = self . next_action oracle , replan_output = self . replan ( oracle , agent_action ) self . next_action , next_subgoal = replan_output if np . argmax ( next_subgoal [ : 4 ] ) == 3 : next_subgoal [ - 2 : ] = 0 self . next_subgoal = next_subgoal if not np . argmax ( next_subgoal [ : 4 ] ) == 0 : self . next_subgoal = next_subgoal self . last_step_error = False self . steps_since_lastfeedback += 1 return oracle
short_docstring: Steps the oracle's internal state forward with the agent's current action.
short_docstring_tokens: Steps the oracle ' s internal state forward with the agent ' s current action .
comment: """
          Steps the oracle's internal state forward with the agent's current action.
          :param agent_action: The action the agent plans to take.
          """$SEP$# If it's explore, mask out the object$SEP$# Don't include open subgoals