hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

[Port Hil-SERL] Add unit tests for the reward classifier & fix imports & check script (#578)

Browse Source
This commit is contained in:
Eugene Mironov
2024-12-23 16:43:55 +07:00
committed by AdilZouitine
parent 66268fcf85
commit 6340d9d17c
6 changed files with 346 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@@ -70,7 +70,9 @@ dependencies = [
"termcolor>=2.4.0",
"torch>=2.2.1",
"torchcodec>=0.2.1; sys_platform != 'win32' and (sys_platform != 'linux' or (platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l'))",
"torchmetrics>=1.6.0",
"torchvision>=0.21.0",
"transformers>=4.47.0",
"wandb>=0.16.3",
"zarr>=2.17.0",
]