1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
|
"""Pymatgen package configuration."""
from __future__ import annotations
import platform
import sys
import numpy as np
from setuptools import Extension, setup
is_win_64 = sys.platform.startswith("win") and platform.machine().endswith("64")
extra_link_args = ["-Wl,--allow-multiple-definition"] if is_win_64 else []
setup(
ext_modules=[
Extension(
"pymatgen.optimization.linear_assignment",
["src/pymatgen/optimization/linear_assignment.pyx"],
extra_link_args=extra_link_args,
),
Extension(
"pymatgen.util.coord_cython",
["src/pymatgen/util/coord_cython.pyx"],
extra_link_args=extra_link_args,
),
Extension(
"pymatgen.optimization.neighbors",
["src/pymatgen/optimization/neighbors.pyx"],
extra_link_args=extra_link_args,
),
],
include_dirs=[np.get_include()],
)
|