⬆️ Raise torch security floors and fold the extras that restate them

Signed-off-by: Onuralp SEZER <thunderbirdtr@gmail.com>

Author: onuralpszr

pyproject.toml

@@ -79,7 +79,13 @@ build = [
 ]
 
 [project.optional-dependencies]
-torch = ["torch>=2.4.1"]
+# torch shipped its last cp38 wheel in 2.4.1 and its last cp39 wheel in 2.8.0,
+# so each interpreter floors at the newest release it still has wheels for
+torch = [
+  "torch>=2.13.0;python_version>='3.10'",
+  "torch>=2.8.0;python_version=='3.9'",
+  "torch>=2.4.1;python_version<'3.9'",
+]
 yolov5 = ["yolov5>=6.0.0,<8.0.0"]
 ultralytics = ["ultralytics>=8.0.0"]
 transformers = ["transformers>=4.49.0;python_version>='3.9'"]
@@ -94,47 +100,19 @@ onnx = [
   "onnxruntime>=1.23.2;python_version>='3.14'",
 ]
 numba = ["numba>=0.59.0;python_version>='3.10'"]
-torchvision = ["torch>=2.4.1", "torchvision>=0.19.0"]
-all = [
-  "yolov5>=6.0.0,<8.0.0",
-  "ultralytics>=8.0.0",
-  "torch>=2.4.1",
-  "torchvision",
-  "transformers>=4.49.0;python_version>='3.9'",
-  # inference pins onnxruntime<1.22, which has no cp314 wheel
-  "inference>=0.51.5;python_version>='3.12' and python_version<'3.14'",
-  "rfdetr>=1.6.2;python_version>='3.12' and python_version<'3.14'",
-  "onnx;python_version>='3.10'",
-  "onnxruntime;python_version>='3.10' and python_version<'3.14'",
-  "onnxruntime>=1.23.2;python_version>='3.14'",
-]
+torchvision = ["sahi[torch]", "torchvision>=0.19.0"]
+all = ["sahi[yolov5,ultralytics,torchvision,transformers,roboflow,onnx]"]
 
 ci = [
-  # max supported pytorch version for python 3.8 is 2.4.1, and 2.8.0 ships no cp314 wheel
-  # CPU versions for Linux/Windows CI
-  "torch==2.9.0+cpu;python_version>='3.14' and platform_system!='Darwin'",
-  "torch==2.8.0+cpu;python_version>='3.9' and python_version<'3.14' and platform_system!='Darwin'",
-  "torch==2.4.1+cpu;python_version=='3.8' and platform_system!='Darwin'",
-  "torchvision==0.24.0+cpu;python_version>='3.14' and platform_system!='Darwin'",
-  "torchvision==0.23.0;python_version>='3.9' and python_version<'3.14' and platform_system!='Darwin'",
-  "torchvision==0.19.1+cpu;python_version=='3.8' and platform_system!='Darwin'",
-  # Regular versions for macOS
-  "torch==2.9.0;python_version>='3.14' and platform_system=='Darwin'",
-  "torch==2.8.0;python_version>='3.9' and python_version<'3.14' and platform_system=='Darwin'",
-  "torch==2.4.1;python_version=='3.8' and platform_system=='Darwin'",
-  "torchvision==0.24.0;python_version>='3.14' and platform_system=='Darwin'",
-  "torchvision==0.23.0;python_version>='3.9' and python_version<'3.14' and platform_system=='Darwin'",
-  "torchvision==0.19.1;python_version=='3.8' and platform_system=='Darwin'",
-  # transformers is supported for python>=3.9
-  "transformers>=4.49.0;python_version>='3.9'",
-  # test roboflow only for python>=3.12; inference pins onnxruntime<1.22, which has no cp314 wheel
-  "inference>=0.51.5;python_version>='3.12' and python_version<'3.14'",
-  "rfdetr>=1.6.2;python_version>='3.12' and python_version<'3.14'",
-  # ultralytics onnx is supported for python>=3.10
-  "onnx;python_version>='3.10'",
-  "onnxruntime;python_version>='3.10' and python_version<'3.14'",
-  "onnxruntime>=1.23.2;python_version>='3.14'",
-  # These are available for all python versions
+  "sahi[transformers,roboflow,onnx]",
+  # pin the floors the torch extra declares; a bare version also matches the
+  # +cpu local build that the pytorch-cpu index serves Linux and Windows
+  "torch==2.13.0;python_version>='3.10'",
+  "torch==2.8.0;python_version=='3.9'",
+  "torch==2.4.1;python_version=='3.8'",
+  "torchvision==0.28.0;python_version>='3.10'",
+  "torchvision==0.23.0;python_version=='3.9'",
+  "torchvision==0.19.1;python_version=='3.8'",
   "pycocotools>=2.0.7",
   "ultralytics>=8.3.161",
   "scikit-image",