hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

revision 1

Browse Source
This commit is contained in:
Alexander Soare
2024-04-15 10:56:43 +01:00
parent 40d417ef60
commit 30023535f9
6 changed files with 24 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_available.py
View File

@@ -18,14 +18,14 @@ from lerobot.common.datasets.xarm import XarmDataset
from lerobot.common.datasets.aloha import AlohaDataset
from lerobot.common.datasets.pusht import PushtDataset
from lerobot.common.policies.act.modeling_act import ActPolicy
from lerobot.common.policies.act.modeling_act import ActionChunkingTransformerPolicy
from lerobot.common.policies.diffusion.policy import DiffusionPolicy
from lerobot.common.policies.tdmpc.policy import TDMPCPolicy
def test_available():
policy_classes = [
ActPolicy,
ActionChunkingTransformerPolicy,
DiffusionPolicy,
TDMPCPolicy,
]