From f5f47ae74311b6e8beebe3c24181b47529474058 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Mon, 1 Sep 2025 10:03:26 +0000
Subject: [PATCH 1/3] Initial plan


From 9ebbdfc3c9ec5c4f973f24f5211ffd45c841c2d8 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Mon, 1 Sep 2025 10:11:27 +0000
Subject: [PATCH 2/3] Update torch to 2.8.0 and modernize dependencies to fix
 security alerts

Co-authored-by: jmercat <11698893+jmercat@users.noreply.github.com>
---
 requirements.txt | 10 +++++-----
 setup.py         | 12 ++++++------
 2 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 53f5496..ee5804b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,10 +1,10 @@
-# need to install torch==1.13.1+cu117 separately with  `pip install torch==1.13.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117`
+# torch>=2.1.0 should be installed separately with specific index if CUDA support needed: `pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118`
 matplotlib
 numpy==1.26.4
 opencv-python
-mmcv==1.4.7
+mmcv-lite
 yapf==0.32.0
-pytorch-lightning==1.7.7
+lightning>=2.1.0
 pytest
 setuptools>=59.5.0
 wandb
@@ -12,7 +12,7 @@ plotly
 scipy
 gradio
 einops
-pydantic==1.10
+pydantic>=2.0.0
 fire
 packaging
-torchmetrics==0.11.4
\ No newline at end of file
+torchmetrics>=1.0.0
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 41581bc..6808cd7 100644
--- a/setup.py
+++ b/setup.py
@@ -10,19 +10,19 @@
     packages=["risk_biased"],
     zip_safe=False,
     install_requires=[
-        "torch==1.13.1+cu117",
+        "torch>=2.1.0",
         "matplotlib",
-        "numpy==1.26.4",
-        "mmcv==1.4.7",
-        "pytorch-lightning==1.7.7",
+        "numpy==1.26.4", 
+        "mmcv-lite",  # Use mmcv-lite instead of mmcv for compatibility
+        "lightning>=2.1.0",  # Using lightning instead of pytorch-lightning
         "pytest",
         "setuptools>=59.5.0",
         "wandb",
         "plotly",
         "scipy",
-        "pydantic==1.10",
+        "pydantic>=2.0.0",
         "gradio",
         "einops",
-        "torchmetrics==0.11.4"
+        "torchmetrics>=1.0.0"
     ],
 )

From 2e626e7189aa432f12da6b093ec18c31a9219a95 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Mon, 15 Sep 2025 16:49:54 +0000
Subject: [PATCH 3/3] Fix pytorch-lightning compatibility by using 1.8.6+
 instead of 2.x

Co-authored-by: jmercat <11698893+jmercat@users.noreply.github.com>
---
 requirements.txt | 10 +++++-----
 setup.py         | 12 ++++++------
 2 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index ee5804b..7057ca9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,10 +1,10 @@
-# torch>=2.1.0 should be installed separately with specific index if CUDA support needed: `pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118`
+# torch>=1.13.1 should be installed separately with specific index if CUDA support needed: `pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118`
 matplotlib
 numpy==1.26.4
 opencv-python
-mmcv-lite
+mmcv==1.4.7
 yapf==0.32.0
-lightning>=2.1.0
+pytorch-lightning>=1.8.6,<2.0.0
 pytest
 setuptools>=59.5.0
 wandb
@@ -12,7 +12,7 @@ plotly
 scipy
 gradio
 einops
-pydantic>=2.0.0
+pydantic>=1.10,<2.0.0
 fire
 packaging
-torchmetrics>=1.0.0
\ No newline at end of file
+torchmetrics>=0.11.4,<1.0.0
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 6808cd7..08ee550 100644
--- a/setup.py
+++ b/setup.py
@@ -10,19 +10,19 @@
     packages=["risk_biased"],
     zip_safe=False,
     install_requires=[
-        "torch>=2.1.0",
+        "torch>=1.13.1",
         "matplotlib",
-        "numpy==1.26.4", 
-        "mmcv-lite",  # Use mmcv-lite instead of mmcv for compatibility
-        "lightning>=2.1.0",  # Using lightning instead of pytorch-lightning
+        "numpy==1.26.4",
+        "mmcv==1.4.7",
+        "pytorch-lightning>=1.8.6,<2.0.0",
         "pytest",
         "setuptools>=59.5.0",
         "wandb",
         "plotly",
         "scipy",
-        "pydantic>=2.0.0",
+        "pydantic>=1.10,<2.0.0",
         "gradio",
         "einops",
-        "torchmetrics>=1.0.0"
+        "torchmetrics>=0.11.4,<1.0.0"
     ],
 )