Merge pull request #61 from macaodha/feat/loosen-dependencies

Remove version cap for torch and torchaudio, add Python 3.13/3.14 support

.github/workflows/python-package.yml

@@ -12,7 +12,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.9", "3.10", "3.11", "3.12"]
+        python-version: ["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]
 
     steps:
       - uses: actions/checkout@v4

pyproject.toml

@@ -14,11 +14,12 @@ dependencies = [
   "pandas>=1.5.3",
   "scikit-learn>=1.2.2",
   "scipy>=1.10.1",
-    "torch>=1.13.1,<2.5.0",
-    "torchaudio>=1.13.1,<2.5.0",
+  "torch>=1.13.1",
+  "torchaudio>=1.13.1",
   "torchvision>=0.14.0",
+  "setuptools>=70.1.0; python_version == '3.14'",
 ]
-requires-python = ">=3.9,<3.13"
+requires-python = ">=3.9,<3.15"
 readme = "README.md"
 license = { text = "CC-by-nc-4" }
 classifiers = [
@@ -30,6 +31,8 @@ classifiers = [
   "Programming Language :: Python :: 3.10",
   "Programming Language :: Python :: 3.11",
   "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14",
   "Topic :: Scientific/Engineering :: Artificial Intelligence",
   "Topic :: Software Development :: Libraries :: Python Modules",
   "Topic :: Multimedia :: Sound/Audio :: Analysis",
@@ -51,8 +54,8 @@ build-backend = "hatchling.build"
 [project.scripts]
 batdetect2 = "batdetect2.cli:cli"
 
-[tool.uv]
-dev-dependencies = [
+[dependency-groups]
+dev = [
   "debugpy>=1.8.8",
   "hypothesis>=6.118.7",
   "pyright>=1.1.388",
@@ -60,6 +63,30 @@ dev-dependencies = [
   "ruff>=0.7.3",
 ]
 
+[[tool.uv.index]]
+name = "pytorch-cu126"
+url = "https://download.pytorch.org/whl/cu126"
+explicit = true
+
+[[tool.uv.index]]
+name = "pytorch-cu118"
+url = "https://download.pytorch.org/whl/cu118"
+explicit = true
+
+[tool.uv.sources]
+torch = [
+  { index = "pytorch-cu118", marker = "python_version == '3.9' and (sys_platform == 'linux' or sys_platform == 'win32')" },
+  { index = "pytorch-cu126", marker = "python_version != '3.9' and (sys_platform == 'linux' or sys_platform == 'win32')" },
+]
+torchvision = [
+  { index = "pytorch-cu118", marker = "python_version == '3.9' and (sys_platform == 'linux' or sys_platform == 'win32')" },
+  { index = "pytorch-cu126", marker = "python_version != '3.9' and (sys_platform == 'linux' or sys_platform == 'win32')" },
+]
+torchaudio = [
+  { index = "pytorch-cu118", marker = "python_version == '3.9' and (sys_platform == 'linux' or sys_platform == 'win32')" },
+  { index = "pytorch-cu126", marker = "python_version != '3.9' and (sys_platform == 'linux' or sys_platform == 'win32')" },
+]
+
 [tool.ruff]
 line-length = 79
 target-version = "py39"