mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-31 19:01:28 +00:00
18 lines
437 B
Python
18 lines
437 B
Python
import abc
|
|
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
|
|
import draccus
|
|
|
|
|
|
@dataclass(kw_only=True)
|
|
class RobotConfig(draccus.ChoiceRegistry, abc.ABC):
|
|
# Allows to distinguish between different robots of the same type
|
|
id: str | None = None
|
|
# Directory to store calibration file
|
|
calibration_dir: Path | None = None
|
|
|
|
@property
|
|
def type(self) -> str:
|
|
return self.get_choice_name(self.__class__)
|