From 8a9be8530653b466da310b486176821bbfdbb256 Mon Sep 17 00:00:00 2001 From: nemo Date: Mon, 14 Jul 2025 16:44:58 +0200 Subject: [PATCH] Fix failing tests --- src/lerobot/record.py | 2 +- tests/utils/test_train_utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lerobot/record.py b/src/lerobot/record.py index 7a607d3c1..9cfbcad2b 100644 --- a/src/lerobot/record.py +++ b/src/lerobot/record.py @@ -305,7 +305,7 @@ def record(cfg: RecordConfig) -> LeRobotDataset: # Load pretrained policy - if cfg.policy.use_peft: + if cfg.policy and cfg.policy.use_peft: from peft import PeftModel logging.info("Loading policy's PEFT adapter.") diff --git a/tests/utils/test_train_utils.py b/tests/utils/test_train_utils.py index 1fc02db82..312b1398c 100644 --- a/tests/utils/test_train_utils.py +++ b/tests/utils/test_train_utils.py @@ -79,7 +79,7 @@ def test_save_checkpoint(mock_save_training_state, tmp_path, optimizer): 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): policy = Mock() policy.config = Mock()