fixes for merging

This commit is contained in:
Dana
2025-06-03 14:29:56 +02:00
parent 31c30cce82
commit 812feac7d7
3 changed files with 7 additions and 2 deletions

View File

@@ -23,6 +23,7 @@ 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 lerobot.common.policies.smolvla.modeling_smolvla import SmolVLAPolicy
from tests.utils import require_env
@@ -50,6 +51,7 @@ def test_available_policies():
DiffusionPolicy,
TDMPCPolicy,
VQBeTPolicy,
SmolVLAPolicy
]
policies = [pol_cls.name for pol_cls in policy_classes]
assert set(policies) == set(lerobot.available_policies), policies