tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Update readme & remove example 1 (#108)

Browse Source 
Co-authored-by: Remi <re.cadene@gmail.com>
- Update instructions for installing the library
- Remove deprecated example 1 (as we are now only using `LeRobotDataset` since #91)
This commit is contained in:
Simon Alibert
2024-04-27 09:48:02 +02:00
committed by GitHub
parent fe2b9af64f
commit bf2eebb090
7 changed files with 28 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/test_examples.py
View File

@@ -16,24 +16,18 @@ def _run_script(path):
def test_example_1():
path = "examples/1_load_hugging_face_dataset.py"
path = "examples/1_load_lerobot_dataset.py"
_run_script(path)
assert Path("outputs/examples/1_load_hugging_face_dataset/episode_5.mp4").exists()
assert Path("outputs/examples/1_load_lerobot_dataset/episode_5.mp4").exists()
def test_example_2():
path = "examples/2_load_lerobot_dataset.py"
_run_script(path)
assert Path("outputs/examples/2_load_lerobot_dataset/episode_5.mp4").exists()
def test_examples_4_and_3():
def test_examples_3_and_2():
"""
Train a model with example 3, check the outputs.
Evaluate the trained model with example 2, check the outputs.
"""
path = "examples/4_train_policy.py"
path = "examples/3_train_policy.py"
with open(path) as file:
file_contents = file.read()
@@ -55,7 +49,7 @@ def test_examples_4_and_3():
for file_name in ["model.pt", "config.yaml"]:
assert Path(f"outputs/train/example_pusht_diffusion/{file_name}").exists()
path = "examples/3_evaluate_pretrained_policy.py"
path = "examples/2_evaluate_pretrained_policy.py"
with open(path) as file:
file_contents = file.read()