add rotary cartpole env

This commit is contained in:
2026-03-08 22:58:32 +01:00
parent c8f28ffbcc
commit c753c369b4
15 changed files with 464 additions and 171 deletions

View File

@@ -57,3 +57,10 @@ class BaseEnv(abc.ABC, Generic[T]):
def compute_truncations(self, step_counts: torch.Tensor) -> torch.Tensor:
return step_counts >= self.config.max_steps
def get_default_qpos(self, nq: int) -> list[float] | None:
"""Return the default joint positions for reset.
Override in subclass if the URDF zero pose doesn't match
the desired initial state (e.g. pendulum hanging down).
Returns None to use the URDF default (all zeros)."""
return None