build: bump python 3.13

pyproject.toml

@@ -103,7 +103,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",
@@ -156,7 +156,6 @@ dependencies = [
     "peft>=0.18.0,<0.19.0",
     "trl<=0.21.0",
     "termcolor==2.3.0",
-    "realesrgan"
 ]
 
 [project.optional-dependencies]
@@ -195,6 +194,9 @@ awq = [
     "llmcompressor>=0.9",
     "torch>=2.9.0"
 ]
+upscale = [
+    "realesrgan",
+]
 full = [
     "pruna[stable-fast]",
 ]

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:
         """

tests/algorithms/testers/upscale.py

@@ -1,9 +1,11 @@
+import pytest
+
 from pruna.algorithms.upscale import RealESRGAN
 
 from .base_tester import AlgorithmTesterBase
 
 
-# Takes too long to run on CPU, so we explicitly exclude it
+@pytest.mark.requires_upscale
 class TestUpscale(AlgorithmTesterBase):
     """Test the Upscale algorithm."""
 

tests/conftest.py

@@ -23,6 +23,7 @@ def pytest_configure(config: Any) -> None:
     config.addinivalue_line("markers", "requires_intel: mark test that needs pruna[intel]")
     config.addinivalue_line("markers", "requires_lmharness: mark test that needs pruna[lmharness]")
     config.addinivalue_line("markers", "requires_whisper: mark test that needs pruna[whisper]")
+    config.addinivalue_line("markers", "requires_upscale: mark test that needs pruna[upscale]")
     config.addinivalue_line("markers", "requires_rapidata: mark test that needs pruna[rapidata]")
     # Category marks
     config.addinivalue_line("markers", "slow: mark test that run rather long")