From ac9ee697f6dbb598700993eb3648bc57de4aa46e Mon Sep 17 00:00:00 2001
From: Gaspar Rochette <gaspar.rochette@pruna.ai>
Date: Thu, 9 Apr 2026 14:55:50 +0000
Subject: [PATCH 1/2] build: bump max python to 3.13

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 5626c3d8..f95ecea2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -108,7 +108,7 @@ authors = [
 ]
 license = {file = "LICENSE"}
 readme = "README.md"
-requires-python = ">=3.10,<3.13"
+requires-python = ">=3.10,<3.14"
 keywords = ["AI", "machine learning", "model optimization", "pruning"]
 classifiers = [
     "Development Status :: 4 - Beta",

From 08ed0faf85655f5cd80927b7835e15e495608586 Mon Sep 17 00:00:00 2001
From: Gaspar Rochette <gaspar.rochette@pruna.ai>
Date: Fri, 10 Apr 2026 12:20:41 +0000
Subject: [PATCH 2/2] build: isolate realesrgan in a extra for python 3.13
 compatibility

---
 pyproject.toml                  | 4 +++-
 src/pruna/algorithms/upscale.py | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index f95ecea2..736f546a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -161,7 +161,6 @@ dependencies = [
     "peft>=0.18.0",
     "trl<=0.21.0",
     "termcolor==2.3.0",
-    "realesrgan"
 ]
 
 [project.optional-dependencies]
@@ -193,6 +192,9 @@ awq = [
     "llmcompressor>=0.9",
     "torch>=2.9.0"
 ]
+upscale = [
+    "realesrgan",
+]
 full = [
     "pruna[stable-fast]",
 ]
diff --git a/src/pruna/algorithms/upscale.py b/src/pruna/algorithms/upscale.py
index da28224c..b3a848a2 100644
--- a/src/pruna/algorithms/upscale.py
+++ b/src/pruna/algorithms/upscale.py
@@ -80,6 +80,7 @@ class RealESRGAN(PrunaAlgorithmBase):
         "ring_attn",
         "hyper",
     ]
+    required_install: str = "``pip install pruna[upscale]``"
 
     def get_hyperparameters(self) -> list:
         """