File size: 1,365 Bytes
b2cec57
 
 
 
 
 
 
 
 
0c3d8ae
f5992a2
b2cec57
0c3d8ae
 
6408eec
 
45e8b30
8dc20ae
b2cec57
 
7f618bf
 
b2cec57
 
 
 
 
 
f5992a2
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
from pathlib import Path


ROOT = Path(__file__).parents[1]


def test_space_metadata_and_runtime_files():
    readme = (ROOT / "README.md").read_text()
    requirements = (ROOT / "requirements.txt").read_text()
    assert "sdk_version: 5.49.1" in readme
    assert "python_version: '3.10'" in readme
    assert "app_file: app.py" in readme
    assert "gradio==5.49.1" in requirements
    assert "huggingface-hub==0.36.2" in requirements
    assert "torch==2.11.0" in requirements
    assert "torch==2.7.1" not in requirements
    assert "lerobot.git@0cf864870cf29f4738d3ade893e6fd13fbd7cdb5" in requirements
    assert "tqdm-loggable>=0.2" in requirements
    assert (ROOT / "openpi_runtime/openpi/policies/ur_policy.py").is_file()
    assert (ROOT / "openpi_runtime/openpi_client/base_policy.py").is_file()
    assert "`pi05_ur_demo_no_state` (default)" in readme
    assert "`pi05_ur_demo_state`" in readme


def test_vendored_config_contains_state_conditioned_ur_policy():
    source = (ROOT / "openpi_runtime/openpi/training/config.py").read_text()
    assert 'name="pi05_ur_demo_state"' in source
    assert "LeRobotURDataConfig" in source


def test_vendored_download_is_python_310_compatible():
    source = (ROOT / "openpi_runtime/openpi/shared/download.py").read_text()
    assert "datetime.UTC" not in source
    assert "datetime.timezone.utc" in source