Follow transformers single file naming conventions (#124)

2024-05-01 13:09:42 +01:00
parent 986583dc5c
commit 01d5490d44
6 changed files with 62 additions and 58 deletions
--- a/lerobot/common/policies/act/configuration_act.py
+++ b/lerobot/common/policies/act/configuration_act.py
@@ -2,7 +2,7 @@ from dataclasses import dataclass, field


@dataclass
-class ActionChunkingTransformerConfig:
+class ACTConfig:
    """Configuration class for the Action Chunking Transformers policy.

    Defaults are configured for training on bimanual Aloha tasks like "insertion" or "transfer".