From f94092c169502b79cad647a8018e629baf4153dd Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 30 Jul 2025 10:12:01 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- tests/datasets/test_aggregate.py | 3 ++- tests/test_control_robot.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/datasets/test_aggregate.py b/tests/datasets/test_aggregate.py index f1f441e61..4f316f80e 100644 --- a/tests/datasets/test_aggregate.py +++ b/tests/datasets/test_aggregate.py @@ -14,9 +14,10 @@ # See the License for the specific language governing permissions and # limitations under the License. -import torch from unittest.mock import patch +import torch + from lerobot.datasets.aggregate import aggregate_datasets from lerobot.datasets.lerobot_dataset import LeRobotDataset from tests.fixtures.constants import DUMMY_REPO_ID diff --git a/tests/test_control_robot.py b/tests/test_control_robot.py index 8a4663a49..374f98129 100644 --- a/tests/test_control_robot.py +++ b/tests/test_control_robot.py @@ -112,7 +112,7 @@ def test_record_and_replay(tmp_path): ) record(record_cfg) - + # Mock the revision to prevent Hub calls during replay with ( patch("lerobot.datasets.lerobot_dataset.get_safe_version") as mock_get_safe_version,