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

fix test imports

Browse Source
This commit is contained in:
Steven Palma
2026-04-11 21:07:53 +02:00
parent c9636bb53f
commit 5940126fb5
9 changed files with 71 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/utils/test_process.py
+3 -1
View File
@@ -22,7 +22,9 @@ from unittest.mock import patch
import pytest
from lerobot.rl.process import ProcessSignalHandler
pytest.importorskip("grpc")
from lerobot.rl.process import ProcessSignalHandler # noqa: E402
# Fixture to reset shutdown_event_counter and original signal handlers before and after each test
tests/utils/test_replay_buffer.py
+8 -5
View File
@@ -18,12 +18,15 @@ import sys
from collections.abc import Callable
import pytest
import torch
from lerobot.datasets.lerobot_dataset import LeRobotDataset
from lerobot.rl.buffer import BatchTransition, ReplayBuffer, random_crop_vectorized
from lerobot.utils.constants import ACTION, DONE, OBS_IMAGE, OBS_STATE, OBS_STR, REWARD
from tests.fixtures.constants import DUMMY_REPO_ID
pytest.importorskip("grpc")
import torch # noqa: E402
from lerobot.datasets.lerobot_dataset import LeRobotDataset # noqa: E402
from lerobot.rl.buffer import BatchTransition, ReplayBuffer, random_crop_vectorized # noqa: E402
from lerobot.utils.constants import ACTION, DONE, OBS_IMAGE, OBS_STATE, OBS_STR, REWARD # noqa: E402
from tests.fixtures.constants import DUMMY_REPO_ID # noqa: E402
def state_dims() -> list[str]: