env_assembly_bench / errors.py
lukasskellijs's picture
Initial EnvHub package: assembly_bench for Isaac Lab Arena
1e09dca verified
Raw
History Blame Contribute Delete
1.18 kB
"""EnvHub errors for assembly_bench (Isaac Lab Arena)."""
class IsaacLabArenaError(RuntimeError):
"""Base exception for IsaacLab Arena environment errors."""
def __init__(self, message: str = "IsaacLab Arena error"):
self.message = message
super().__init__(self.message)
class IsaacLabArenaConfigError(IsaacLabArenaError):
"""Invalid observation / action configuration against the live env."""
def __init__(self, invalid: list, available: list, key_type: str = "keys"):
msg = f"Invalid {key_type}: {invalid}. Available: {sorted(available)}"
super().__init__(msg)
self.invalid = invalid
self.available = available
class IsaacLabArenaCameraKeyError(IsaacLabArenaConfigError):
"""camera_keys do not match available camera observation terms."""
def __init__(self, invalid: list, available: list):
super().__init__(invalid, available, "camera_keys")
class IsaacLabArenaStateKeyError(IsaacLabArenaConfigError):
"""state_keys do not match available policy observation terms."""
def __init__(self, invalid: list, available: list):
super().__init__(invalid, available, "state_keys")