Add VQ-BeT (#166)

2024-06-26 16:55:02 +09:00
parent 9aa4cdb976
commit 7d1542cae1
9 changed files with 2644 additions and 1 deletions
--- a/tests/test_available.py
+++ b/tests/test_available.py
@@ -22,6 +22,7 @@ import lerobot
 from lerobot.common.policies.act.modeling_act import ACTPolicy
 from lerobot.common.policies.diffusion.modeling_diffusion import DiffusionPolicy
 from lerobot.common.policies.tdmpc.modeling_tdmpc import TDMPCPolicy
+from lerobot.common.policies.vqbet.modeling_vqbet import VQBeTPolicy
 from tests.utils import require_env


@@ -48,6 +49,7 @@ def test_available_policies():
        ACTPolicy,
        DiffusionPolicy,
        TDMPCPolicy,
+        VQBeTPolicy,
    ]
    policies = [pol_cls.name for pol_cls in policy_classes]
    assert set(policies) == set(lerobot.available_policies), policies