From ee87c5b9bac3dbc64f60a8c77c20a8b941ce5657 Mon Sep 17 00:00:00 2001 From: Francesco Parisio Date: Tue, 2 Dec 2025 11:11:41 +0100 Subject: [PATCH 1/2] bug: fix semver comparison logic using packaging.version.Version --- setup.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 0dde1e4..367f574 100644 --- a/setup.py +++ b/setup.py @@ -8,6 +8,7 @@ import glob from setuptools import find_packages, setup +from packaging.version import Version from torch.utils.cpp_extension import ( CppExtension, @@ -18,10 +19,7 @@ library_name = "extension_cpp" -if torch.__version__ >= "2.6.0": - py_limited_api = True -else: - py_limited_api = False +py_limited_api = Version(torch.__version__) >= Version("2.6.0") def get_extensions(): From b5e7563a0a043032fd074d2f3c2282696fb1fc5a Mon Sep 17 00:00:00 2001 From: Francesco Parisio Date: Tue, 2 Dec 2025 11:30:02 +0100 Subject: [PATCH 2/2] fix: add packaging to requirements --- requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements.txt b/requirements.txt index af3149e..f9c596d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,3 @@ torch numpy +packaging