def METHOD_NAME(
    self, action: WrapperActType
) -> tuple[WrapperObsType, SupportsFloat, bool, bool, dict]:
    """Transforms the action to a jax array .
    Args:
        action: the action to perform as a numpy array
    Returns:
        A tuple containing numpy versions of the next observation, reward, termination, truncation, and extra info.
    """
    jax_action = numpy_to_jax(action)
    obs, reward, terminated, truncated, info = self.env.METHOD_NAME(jax_action)
    return (
        jax_to_numpy(obs),
        float(reward),
        bool(terminated),
        bool(truncated),
        jax_to_numpy(info),
    )