class_docstr: Interface for all the EvoJAX tasks.
docstr: This steps once the simulation.
  
  Args:
      state - System internal states of shape (num_tasks, *).
      action - Vectorized actions of shape (num_tasks, action_size).
  Returns:
      TaskState. Task states.
      jnp.ndarray. Reward.
      jnp.ndarray. Task termination flag: 1 for done, 0 otherwise.
docstr_tok: ["This","steps","once","the","simulation",".","Args",":","state","-","System","internal","states","of","shape","(","num_tasks",",","*",")",".","action","-","Vectorized","actions","of","shape","(","num_tasks",",","action_size",")",".","Returns",":","TaskState",".","Task","states",".","jnp.ndarray",".","Reward",".","jnp.ndarray",".","Task","termination","flag",":","1","for","done",",","0","otherwise","."]
code: def step(
      self, state: TaskState, action: jnp.ndarray
  ) -> Tuple[TaskState, jnp.ndarray, jnp.ndarray]:
      """This steps once the simulation.
  
      Args:
          state - System internal states of shape (num_tasks, *).
          action - Vectorized actions of shape (num_tasks, action_size).
      Returns:
          TaskState. Task states.
          jnp.ndarray. Reward.
          jnp.ndarray. Task termination flag: 1 for done, 0 otherwise.
      """
      raise NotImplementedError()
  
imports: from abc import ABC
  from abc import abstractmethod
  from typing import Dict
  from typing import List
  from typing import Tuple
  from typing import TypeVar
  import dataclasses
  import jax.numpy
  from flax.struct import dataclass
label_desc: Use image node_id 2 for calling the VectorizedTask obj's underlying member method code with example usage: obj.step(state, action) without return types