11import versioneer
22from packaging .version import Version
33import os
4-
5- try :
6- import importlib .metadata as metadata
7- get_version = lambda x : metadata .version (x )
8- PkgNotFound = metadata .PackageNotFoundError
9- except ImportError :
10- import pkg_resources
11- get_version = lambda x : pkg_resources .get_distribution (x ).version
12- PkgNotFound = pkg_resources .DistributionNotFound
13-
144from setuptools import setup , find_packages
155
166
@@ -25,21 +15,21 @@ def min_max(pkgs, pkg_name):
2515
2616def numpy_compat (required ):
2717 new_reqs = [r for r in required if "numpy" not in r and "sympy" not in r ]
28- sympy_lb , sympy_ub = min_max (required , "sympy" )
2918 numpy_lb , numpy_ub = min_max (required , "numpy" )
3019
3120 # Due to api changes in numpy 2.0, it requires sympy 1.12.1 at the minimum
3221 # Check if sympy is installed and enforce numpy version accordingly.
3322 # If sympy isn't installed, enforce sympy>=1.12.1 and numpy>=2.0
34- try :
35- sympy_version = Version (get_version ("sympy" ))
36- min_ver2 = Version ("1.12.1" )
37- if sympy_version < min_ver2 :
38- new_reqs .extend ([f"numpy>{ numpy_lb } ,<2.0" , f"sympy=={ sympy_version } " ])
39- else :
40- new_reqs .extend ([f"numpy>=2.0,<{ numpy_ub } " , f"sympy=={ sympy_version } " ])
41- except PkgNotFound :
42- new_reqs .extend ([f"sympy>=1.12.1,<{ sympy_ub } " , f"numpy>=2.0,<{ numpy_ub } " ])
23+ # try:
24+ import sympy
25+ sympy_version = Version (sympy .__version__ )
26+ min_ver2 = Version ("1.12.1" )
27+ if sympy_version < min_ver2 :
28+ new_reqs .extend ([f"numpy>{ numpy_lb } ,<2.0" , f"sympy=={ sympy_version } " ])
29+ else :
30+ new_reqs .extend ([f"numpy>=2.0,<{ numpy_ub } " , f"sympy=={ sympy_version } " ])
31+ # except ImportError:
32+ # new_reqs.extend([f"sympy>=1.12.1,<{sympy_ub}", f"numpy>=2.0,<{numpy_ub}"])
4333
4434 return new_reqs
4535
0 commit comments