update to pytorch 2.4

Author: Hananel-Hazan

pyproject.toml

@@ -15,30 +15,30 @@ numpy = "^2"
 scipy = "^1"
 Cython = "^3"
 torch = [
-     {version = "2.3.1", platform = "linux", source = "torch+cu121"},
-     {version = "2.3.1", platform = "darwin"},
-     {version = "2.3.1", platform = "win32", source = "torch+cu121"},
-     {version = "2.3.1", platform = "win_amd64", source = "torch+cu121"},
+     {version = "2.4.0", platform = "linux", source = "torch+cu124"},
+     {version = "2.4.0", platform = "darwin"},
+     {version = "2.4.0", platform = "win32", source = "torch+cu124"},
+     {version = "2.4.0", platform = "win_amd64", source = "torch+cu124"},
  ]
 torchvision = [
-     {version = "0.18.1", platform = "darwin"},
-     {version = "0.18.1", platform = "linux", source = "torch+cu121"},
-     {version = "0.18.1", platform = "win32", source = "torch+cu121"},
-     {version = "0.18.1", platform = "win_amd64", source = "torch+cu121"},
+     {version = "0.19.0", platform = "darwin"},
+     {version = "0.19.0", platform = "linux", source = "torch+cu124"},
+     {version = "0.19.0", platform = "win32", source = "torch+cu124"},
+     {version = "0.19.0", platform = "win_amd64", source = "torch+cu124"},
  ]
 torchaudio = [
-     {version = "2.3.1", platform = "darwin"},
-     {version = "2.3.1", platform = "linux", source = "torch+cu121"},
-     {version = "2.3.1", platform = "win32", source = "torch+cu121"},
-     {version = "2.3.1", platform = "win_amd64", source = "torch+cu121"},
+     {version = "2.4.0", platform = "darwin"},
+     {version = "2.4.0", platform = "linux", source = "torch+cu124"},
+     {version = "2.4.0", platform = "win32", source = "torch+cu124"},
+     {version = "2.4.0", platform = "win_amd64", source = "torch+cu124"},
  ]
 tensorboardX = "^2.6.2"
 tqdm = "^4"
 matplotlib = "^3.9.0"
 gymnasium = {extras = ["atari", "accept-rom-license"], version = "^0.29.1"}
 scikit-build = "^0.18"
 scikit-image = "^0.24"
-scikit-learn = "^1.2"
+scikit-learn = "^1.5"
 opencv-python = "^4"
 pandas = "^2"
 foolbox = "^3"
@@ -53,14 +53,19 @@ foolbox = "^3"
  url = "https://download.pytorch.org/whl/cu121"
  priority = "explicit"
 
+ [[tool.poetry.source]]
+ name = "torch+cu124"
+ url = "https://download.pytorch.org/whl/cu124"
+ priority = "explicit"
+
 [tool.poetry.dev-dependencies]
-pytest = "^7.2.2"
-pre-commit = "^2.14.0"
-notebook = "^7.2"
+pytest = "^8"
+pre-commit = "^3"
+notebook = "^7"
 jupyterlab = "^4"
 isort = "^5.9.3"
-black = "^24.1.0"
-autoflake = "^2.0.1"
+black = "^24"
+autoflake = "^2"
 
 [build-system]
 requires = ["setuptools", "poetry-core>=1.0.0"]