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

Fix failing tests

Browse Source
This commit is contained in:
nemo
2025-07-14 16:44:58 +02:00
parent 57014e5bb1
commit 8a9be85306
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/record.py
+1 -1
View File
@@ -305,7 +305,7 @@ def record(cfg: RecordConfig) -> LeRobotDataset:
# Load pretrained policy # Load pretrained policy
if cfg.policy.use_peft: if cfg.policy and cfg.policy.use_peft:
from peft import PeftModel from peft import PeftModel
logging.info("Loading policy's PEFT adapter.") logging.info("Loading policy's PEFT adapter.")
tests/utils/test_train_utils.py
+1 -1
View File
@@ -79,7 +79,7 @@ def test_save_checkpoint(mock_save_training_state, tmp_path, optimizer):
mock_save_training_state.assert_called_once() mock_save_training_state.assert_called_once()
@patch("lerobot.common.utils.train_utils.save_training_state") @patch("lerobot.utils.train_utils.save_training_state")
def test_save_checkpoint_peft(mock_save_training_state, tmp_path, optimizer): def test_save_checkpoint_peft(mock_save_training_state, tmp_path, optimizer):
policy = Mock() policy = Mock()
policy.config = Mock() policy.config = Mock()