From 49ecbeb33f4148c5731eba41e8d827dab41b9dd2 Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Wed, 6 Aug 2025 20:10:47 +0200 Subject: [PATCH] fix(deps): ceil torch pkg versions (#1689) * fix(deps): ceil torch pkg versions * chore(Docs): add todo comment --- pyproject.toml | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index a1db99c24..d26513404 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -68,15 +68,16 @@ dependencies = [ "einops>=0.8.0", "opencv-python-headless>=4.9.0", "av>=14.2.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')", - "torchvision>=0.21.0", "jsonlines>=4.0.0", "packaging>=24.2", "pynput>=1.7.7", "pyserial>=3.5", "wandb>=0.20.0", + "torch>=2.2.1,<2.8.0", # TODO: Bumb dependency + "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 + "torchvision>=0.21.0,<0.23.0", # TODO: Bumb dependency + "draccus==0.10.0", # TODO: Remove == "gymnasium>=0.29.1,<1.0.0", # TODO: Bumb dependency "rerun-sdk>=0.21.0,<0.23.0", # TODO: Bumb dependency