test

pyproject.toml

@@ -10,7 +10,7 @@ version = "0.0.1"
 # The following assumes that we build on top of the NGC PyTorch Docker image.
 dependencies = [
     # MANDATORY for cneuromax/
-    "torch==2.4.1", # Tensor manipulation
+    "torch==2.5.0", # Tensor manipulation
     "hydra-core==1.3.2", # Configuration management
     "hydra-zen==0.13.0", # Cleaner Hydra configs
     "beartype==0.19.0", # Dynamic type checking
@@ -49,8 +49,8 @@ dependencies = [
     # MANDATORY for cneuromax/fitting/deeplearning/
     "lightning==2.4.0", # PyTorch wrapper to decrease boilerplate
     # OPTIONAL for cneuromax/fitting/deeplearning/
-    "torchaudio==2.4.1", # Tensor manipulation on audio data
-    "torchvision==0.19.1", # Tensor manipulation on vision data
+    "torchaudio==2.5.0", # Tensor manipulation on audio data
+    "torchvision==0.20.0", # Tensor manipulation on vision data
     "transformers==4.45.2", # Pre-trained models published on Hugging Face
     "diffusers==0.30.3", # Diffusion models published on Hugging Face
     "timm==1.0.11", # Image models
@@ -88,15 +88,16 @@ dependencies = [
     "pytest-cov==5.0.0", # Test coverage
 ]
 
-[[tool.uv.index]]
-name = "pytorch"
-url = "https://download.pytorch.org/whl/cu124"
-
 [tool.uv.sources]
 torch = { index = "pytorch" }
 torchaudio = { index = "pytorch" }
 torchvision = { index = "pytorch" }
 
+[[tool.uv.index]]
+name = "pytorch"
+url = "https://download.pytorch.org/whl/cu124"
+explicit = true
+
 [tool.black]
 line-length = 79