fix torch min

Author: gsheni

pyproject.toml

@@ -162,6 +162,7 @@ preview = true
 line-length = 100
 indent-width = 4
 src = ["ctgan"]
+target-version = "py314"
 exclude = [
     "docs",
     ".tox",
@@ -193,7 +194,6 @@ ignore = [
     # pydocstyle
     "D107",  # Missing docstring in __init__
     "D417",   # Missing argument descriptions in the docstring, this is a bug from pydocstyle: https://github.com/PyCQA/pydocstyle/issues/449
-    "PD901",
     "PD101",
 ]
 

tasks.py

@@ -1,6 +1,7 @@
 import inspect
 import operator
 import os
+import platform
 import shutil
 import stat
 import sys
@@ -12,6 +13,7 @@
 from packaging.version import Version
 
 COMPARISONS = {'>=': operator.ge, '>': operator.gt, '<': operator.lt, '<=': operator.le}
+EXTERNAL_DEPENDENCY_CAPS_FOR_WINDOWS = {'torch': '2.9.0'}
 
 if not hasattr(inspect, 'getargspec'):
     inspect.getargspec = inspect.getfullargspec
@@ -74,6 +76,8 @@ def _get_minimum_versions(dependencies, python_version):
                 (spec.version for spec in req.specifier if spec.operator in ('>=', '==')),
                 existing_version,
             )
+            if isinstance(new_version, str):
+                new_version = Version(new_version)
             if new_version > existing_version:
                 min_versions[req.name] = (
                     f'{req.name}=={new_version}'  # Change when a valid newer version is found
@@ -94,8 +98,9 @@ def install_minimum(c):
     if minimum_versions:
         install_deps = ' '.join(minimum_versions)
         c.run(f'python -m pip install {install_deps}')
-
-
+        if platform.system() == 'Windows' and sys.version_info < (3,14):
+            for dep, cap in EXTERNAL_DEPENDENCY_CAPS_FOR_WINDOWS.items():
+                c.run(f'python -m pip install "{dep}<{cap}"')
 @task
 def minimum(c):
     install_minimum(c)