wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added todos

This commit is contained in:
Simon Alibert 2024-04-20 14:50:58 +02:00
parent 7e3f284e67
commit 8a0d847de7
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_datasets.py
View File

@ -92,6 +92,7 @@ def test_compute_stats_on_xarm():
We compare with taking a straight min, mean, max, std of all the data in one pass (which we can do We compare with taking a straight min, mean, max, std of all the data in one pass (which we can do
because we are working with a small dataset). because we are working with a small dataset).
""" """
# TODO(rcadene): Reduce size of dataset sample on which stats compute is tested
from lerobot.common.datasets.xarm import XarmDataset from lerobot.common.datasets.xarm import XarmDataset
data_dir = Path(os.environ["DATA_DIR"]) if "DATA_DIR" in os.environ else None data_dir = Path(os.environ["DATA_DIR"]) if "DATA_DIR" in os.environ else None

1
tests/test_examples.py
View File

@ -1,3 +1,4 @@
# TODO(aliberts): Mute logging for these tests
import subprocess import subprocess
from pathlib import Path from pathlib import Path