hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

unskip

Browse Source
This commit is contained in:
Cadene
2024-03-26 10:45:31 +00:00
parent 98534d1a63
commit 9ced0cf1fb
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/abstract.py
View File

@@ -70,7 +70,7 @@ class AbstractDataset(TensorDictReplayBuffer):
self.dataset_id = dataset_id self.dataset_id = dataset_id
self.version = version self.version = version
self.shuffle = shuffle self.shuffle = shuffle
self.root = root self.root = root if root is None else Path(root)
if self.root is not None and self.version is not None: if self.root is not None and self.version is not None:
logging.warning( logging.warning(

4
tests/test_examples.py
View File

@@ -1,7 +1,6 @@
import pytest import pytest
from pathlib import Path from pathlib import Path
@pytest.mark.skip(reason="For some reason 1_visualize_dataset.py downloads the dataset")
@pytest.mark.parametrize( @pytest.mark.parametrize(
"path", "path",
[ [
@@ -16,4 +15,5 @@ def test_example(path):
file_contents = file.read() file_contents = file.read()
exec(file_contents) exec(file_contents)
assert Path("outputs/visualize_dataset/example/episode_0.mp4").exists() if path == "examples/1_visualize_dataset.py":
assert Path("outputs/visualize_dataset/example/episode_0.mp4").exists()