Merge remote-tracking branch 'origin/main' into user/aliberts/2025_02_25_refactor_robots

2025-06-03 19:10:17 +02:00
parent 5c2f8ccd14 bfd26eef5a
commit 4687296d93
10 changed files with 1531 additions and 14 deletions
--- a/tests/test_available.py
+++ b/tests/test_available.py
@@ -45,12 +45,7 @@ def test_available_policies():
    This test verifies that the class attribute `name` for all policies is
    consistent with those listed in `lerobot/__init__.py`.
    """
-    policy_classes = [
-        ACTPolicy,
-        DiffusionPolicy,
-        TDMPCPolicy,
-        VQBeTPolicy,
-    ]
+    policy_classes = [ACTPolicy, DiffusionPolicy, TDMPCPolicy, VQBeTPolicy]
    policies = [pol_cls.name for pol_cls in policy_classes]
    assert set(policies) == set(lerobot.available_policies), policies