(self, task: str, obs_type: str = "pixels_agent_pos", seed: int = 0)
| 10 | """An environment for an Aloha robot in simulation.""" |
| 11 | |
| 12 | def __init__(self, task: str, obs_type: str = "pixels_agent_pos", seed: int = 0) -> None: |
| 13 | np.random.seed(seed) |
| 14 | self._rng = np.random.default_rng(seed) |
| 15 | |
| 16 | self._gym = gymnasium.make(task, obs_type=obs_type) |
| 17 | |
| 18 | self._last_obs = None |
| 19 | self._done = True |
| 20 | self._episode_reward = 0.0 |
| 21 | |
| 22 | @override |
| 23 | def reset(self) -> None: |
nothing calls this directly
no outgoing calls
no test coverage detected