class_docstr: Add task_name or task_id to env infos.
  
  Args:
      env (gym.Env): The environment to wrap.
      task_name (str or None): Task name to be added, if any.
      task_id (int or None): Task ID to be added, if any.
class_docstr_tok: ["Add","task_name","or","task_id","to","env","infos",".","Args",":","env","(","gym.Env",")",":","The","environment","to","wrap",".","task_name","(","str","or","None",")",":","Task","name","to","be","added",",","if","any",".","task_id","(","int","or","None",")",":","Task","ID","to","be","added",",","if","any","."]
docstr: gym.Env step for the active task env.
  
  Args:
      action (np.ndarray): Action performed by the agent in the
          environment.
  
  Returns:
      tuple:
          np.ndarray: Agent's observation of the current environment.
          float: Amount of reward yielded by previous action.
          bool: True iff the episode has ended.
          dict[str, np.ndarray]: Contains auxiliary diagnostic
              information about this time-step.
docstr_tok: ["gym.Env","step","for","the","active","task","env",".","Args",":","action","(","np.ndarray",")",":","Action","performed","by","the","agent","in","the","environment",".","Returns",":","tuple",":","np.ndarray",":","Agent","'s","observation","of","the","current","environment",".","float",":","Amount","of","reward","yielded","by","previous","action",".","bool",":","True","iff","the","episode","has","ended",".","dict","[","str",",","np.ndarray","]",":","Contains","auxiliary","diagnostic","information","about","this","time-step","."]
code: def step(self, action):
      """gym.Env step for the active task env.
  
      Args:
          action (np.ndarray): Action performed by the agent in the
              environment.
  
      Returns:
          tuple:
              np.ndarray: Agent's observation of the current environment.
              float: Amount of reward yielded by previous action.
              bool: True iff the episode has ended.
              dict[str, np.ndarray]: Contains auxiliary diagnostic
                  information about this time-step.
  
      """
      es = super().step(action)
      if self._task_name is not None:
          es.env_info["task_name"] = self._task_name
      if self._task_id is not None:
          es.env_info["task_id"] = self._task_id
      return es
  
imports: from garage import Wrapper
label_desc: Use image node_id 2 for calling the TaskNameWrapper obj's underlying member method code with example usage: obj.step(action) and returns: es