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

2024-12-23 16:43:55 +07:00
parent 1a8b99e360
commit 17a3a31b5f
6 changed files with 346 additions and 1 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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')) and (sys_platform != 'darwin' or platform_machine != 'x86_64')",
+    "torchmetrics>=1.6.0",
    "torchvision>=0.21.0",
+    "transformers>=4.47.0",
    "wandb>=0.16.3",
    "zarr>=2.17.0",
 ]