Skip to content

Commit 49ecbeb

Browse files
fix(deps): ceil torch pkg versions (#1689)
* fix(deps): ceil torch pkg versions * chore(Docs): add todo comment
1 parent 88f7bf0 commit 49ecbeb

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pyproject.toml

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -68,15 +68,16 @@ dependencies = [
6868
"einops>=0.8.0",
6969
"opencv-python-headless>=4.9.0",
7070
"av>=14.2.0",
71-
"torch>=2.2.1",
72-
"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')",
73-
"torchvision>=0.21.0",
7471
"jsonlines>=4.0.0",
7572
"packaging>=24.2",
7673
"pynput>=1.7.7",
7774
"pyserial>=3.5",
7875
"wandb>=0.20.0",
7976

77+
"torch>=2.2.1,<2.8.0", # TODO: Bumb dependency
78+
"torchcodec>=0.2.1,<0.6.0; 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')", # TODO: Bumb dependency
79+
"torchvision>=0.21.0,<0.23.0", # TODO: Bumb dependency
80+
8081
"draccus==0.10.0", # TODO: Remove ==
8182
"gymnasium>=0.29.1,<1.0.0", # TODO: Bumb dependency
8283
"rerun-sdk>=0.21.0,<0.23.0", # TODO: Bumb dependency

0 commit comments

Comments
 (0)