admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-24 04:59:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' into feature/add-multitask-dit

Browse Source
This commit is contained in:
Bryson Jones
2026-01-05 08:39:09 -08:00
committed by GitHub
parent e268ec1ec5 6106a8136c
commit 713efb6427
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/wall_x/test_wallx.py
+3 -2
View File
@@ -21,9 +21,10 @@ import os
import pytest import pytest
import torch import torch
# Skip if openpi or transformers is not available # Skip if required dependencies are not available
pytest.importorskip("peft") pytest.importorskip("peft")
pytest.importorskip("transformers==4.49.0") pytest.importorskip("transformers")
pytest.importorskip("torchdiffeq")
# Skip this entire module in CI # Skip this entire module in CI
pytestmark = pytest.mark.skipif( pytestmark = pytest.mark.skipif(
tests/test_cli.py
+1 -1
View File
@@ -208,7 +208,7 @@ def test_peft_record_loads_policy(policy_type, tmp_path):
loaded_policy = kwargs["policy"] loaded_policy = kwargs["policy"]
with ( with (
patch("lerobot.robots.make_robot_from_config", dummy_make_robot_from_config), patch("lerobot.scripts.lerobot_record.make_robot_from_config", dummy_make_robot_from_config),
# disable record loop since we're only interested in successful loading of the policy. # disable record loop since we're only interested in successful loading of the policy.
patch("lerobot.scripts.lerobot_record.record_loop", dummy_record_loop), patch("lerobot.scripts.lerobot_record.record_loop", dummy_record_loop),
# disable speech output # disable speech output