Merge branch 'main' of github.com:huggingface/lerobot into feature/basic-peft-support

tests/utils/test_train_utils.py

@@ -14,7 +14,7 @@
 from pathlib import Path
 from unittest.mock import Mock, patch
 
-from lerobot.common.constants import (
+from lerobot.constants import (
     CHECKPOINTS_DIR,
     LAST_CHECKPOINT_LINK,
     OPTIMIZER_PARAM_GROUPS,
@@ -24,7 +24,7 @@ from lerobot.common.constants import (
     TRAINING_STATE_DIR,
     TRAINING_STEP,
 )
-from lerobot.common.utils.train_utils import (
+from lerobot.utils.train_utils import (
     get_step_checkpoint_dir,
     get_step_identifier,
     load_training_state,
@@ -69,7 +69,7 @@ def test_update_last_checkpoint(tmp_path):
     assert last_checkpoint.resolve() == checkpoint
 
 
-@patch("lerobot.common.utils.train_utils.save_training_state")
+@patch("lerobot.utils.train_utils.save_training_state")
 def test_save_checkpoint(mock_save_training_state, tmp_path, optimizer):
     policy = Mock()
     cfg = Mock()