🧠 Robomimic Adapter

OrcaGym provides a Robomimic adapter for imitation learning.

Environment Adaptation

from orca_gym.adapters.robomimic import robomimic_env
from orca_gym.adapters.robomimic.robomimic_env import RobomimicEnv

class MyTask(RobomimicEnv):
    """Create a Robomimic-compatible environment"""
    def __init__(self, ...):
        super().__init__(...)

Dataset Utilities

from orca_gym.adapters.robomimic import dataset_util

# Process HDF5 datasets

Task Definition

# Custom task class
from orca_gym.adapters.robomimic.task import ...

class MyRobomimicTask(RobomimicEnv):
    # Implement the interface required by Robomimic
    pass