Fix failing tests

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
+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()
@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()