tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Dana
2025-06-03 15:29:10 +02:00
parent 812feac7d7
commit b41362a4aa
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_available.py
View File

@@ -23,7 +23,6 @@ from lerobot.common.policies.act.modeling_act import ACTPolicy
from lerobot.common.policies.diffusion.modeling_diffusion import DiffusionPolicy from lerobot.common.policies.diffusion.modeling_diffusion import DiffusionPolicy
from lerobot.common.policies.tdmpc.modeling_tdmpc import TDMPCPolicy from lerobot.common.policies.tdmpc.modeling_tdmpc import TDMPCPolicy
from lerobot.common.policies.vqbet.modeling_vqbet import VQBeTPolicy from lerobot.common.policies.vqbet.modeling_vqbet import VQBeTPolicy
from lerobot.common.policies.smolvla.modeling_smolvla import SmolVLAPolicy
from tests.utils import require_env from tests.utils import require_env
@@ -51,7 +50,6 @@ def test_available_policies():
DiffusionPolicy, DiffusionPolicy,
TDMPCPolicy, TDMPCPolicy,
VQBeTPolicy, VQBeTPolicy,
SmolVLAPolicy
] ]
policies = [pol_cls.name for pol_cls in policy_classes] policies = [pol_cls.name for pol_cls in policy_classes]
assert set(policies) == set(lerobot.available_policies), policies assert set(policies) == set(lerobot.available_policies), policies