Fix #173 - Require gym-pusht to be installed for test_examples_3_and_2 (#174)

This commit is contained in:
Simon Alibert
2024-05-12 08:08:59 +02:00
committed by GitHub
parent ced3de4c94
commit 29a196c5dd
3 changed files with 53 additions and 1 deletions

View File

@@ -3,6 +3,8 @@ import subprocess
import sys
from pathlib import Path
from tests.utils import require_package
def _find_and_replace(text: str, finds_and_replaces: list[tuple[str, str]]) -> str:
for f, r in finds_and_replaces:
@@ -21,6 +23,7 @@ def test_example_1():
assert Path("outputs/examples/1_load_lerobot_dataset/episode_0.mp4").exists()
@require_package("gym_pusht")
def test_examples_3_and_2():
"""
Train a model with example 3, check the outputs.