XiangpengYang commited on
Commit
8dc20ae
·
1 Parent(s): 0c3d8ae

fix: add OpenPI logging progress dependency

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -0
  2. tests/test_space_config.py +1 -0
requirements.txt CHANGED
@@ -19,6 +19,7 @@ sentencepiece>=0.2.0
19
  torch==2.11.0
20
  typing-extensions>=4.12.2
21
  tyro>=0.9.5
 
22
  wandb>=0.19.1
23
  filelock>=3.16.1
24
  beartype==0.19.0
 
19
  torch==2.11.0
20
  typing-extensions>=4.12.2
21
  tyro>=0.9.5
22
+ tqdm-loggable>=0.2
23
  wandb>=0.19.1
24
  filelock>=3.16.1
25
  beartype==0.19.0
tests/test_space_config.py CHANGED
@@ -15,6 +15,7 @@ def test_space_metadata_and_runtime_files():
15
  assert "torch==2.11.0" in requirements
16
  assert "torch==2.7.1" not in requirements
17
  assert "lerobot.git@0cf864870cf29f4738d3ade893e6fd13fbd7cdb5" in requirements
 
18
  assert (ROOT / "openpi_runtime/openpi/policies/ur_policy.py").is_file()
19
  assert (ROOT / "openpi_runtime/openpi_client/base_policy.py").is_file()
20
 
 
15
  assert "torch==2.11.0" in requirements
16
  assert "torch==2.7.1" not in requirements
17
  assert "lerobot.git@0cf864870cf29f4738d3ade893e6fd13fbd7cdb5" in requirements
18
+ assert "tqdm-loggable>=0.2" in requirements
19
  assert (ROOT / "openpi_runtime/openpi/policies/ur_policy.py").is_file()
20
  assert (ROOT / "openpi_runtime/openpi_client/base_policy.py").is_file()
21