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

fix(tests): gate RL tests on the datasets extra

Browse Source
This commit is contained in:
Khalil Meftah
2026-04-27 16:53:34 +02:00
parent 577f14337a
commit e298474bf3
6 changed files with 50 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/rl/test_data_mixer.py
+8 -4
View File
@@ -13,11 +13,15 @@
# limitations under the License.
"""Tests for RL data mixing (DataMixer, OnlineOfflineMixer)."""
import torch
import pytest
from lerobot.rl.buffer import ReplayBuffer
from lerobot.rl.data_sources import OnlineOfflineMixer
from lerobot.utils.constants import OBS_STATE
pytest.importorskip("datasets", reason="datasets is required (install lerobot[dataset])")
import torch # noqa: E402
from lerobot.rl.buffer import ReplayBuffer # noqa: E402
from lerobot.rl.data_sources import OnlineOfflineMixer # noqa: E402
from lerobot.utils.constants import OBS_STATE # noqa: E402
def _make_buffer(capacity: int = 100, state_dim: int = 4) -> ReplayBuffer: