https://github.com/bashtage/arch
Tip revision: 95d1d6595a09cf118f275625e666e2a9bc077115 authored by Kevin Sheppard on 26 September 2023, 11:01:18 UTC
Merge pull request #681 from bashtage/rls-6.2
Merge pull request #681 from bashtage/rls-6.2
Tip revision: 95d1d65
setup.py
from setuptools import Command, Extension, find_namespace_packages, setup
from setuptools.command.build_ext import build_ext as _build_ext
from setuptools.dist import Distribution
from setuptools.errors import CCompilerError, ExecError, PlatformError
from collections import defaultdict
import fnmatch
import importlib.metadata
import importlib.resources
import os
CYTHON_COVERAGE = os.environ.get("ARCH_CYTHON_COVERAGE", "0") in ("true", "1", "True")
if CYTHON_COVERAGE:
print(
"Building with coverage for cython modules, ARCH_CYTHON_COVERAGE="
+ os.environ["ARCH_CYTHON_COVERAGE"]
)
try:
from Cython.Build import cythonize
CYTHON_INSTALLED = True
except ImportError:
CYTHON_INSTALLED = False
if CYTHON_COVERAGE:
raise ImportError(
"cython is required for cython coverage. Unset " "ARCH_CYTHON_COVERAGE"
)
FAILED_COMPILER_WARNING = """
******************************************************************************
* WARNING *
******************************************************************************
Unable to build binary modules for arch. While these are not required to run
any code in the package, it is strongly recommended to either compile the
extension modules or to use numba.
******************************************************************************
* WARNING *
******************************************************************************
"""
# prevent setup.py from crashing by calling import numpy before numpy is installed
class build_ext(_build_ext):
def build_extensions(self) -> None:
ref = importlib.resources.files("numpy") / "core/include"
with importlib.resources.as_file(ref) as path:
numpy_incl = str(path)
for ext in self.extensions:
if hasattr(ext, "include_dirs") and numpy_incl not in ext.include_dirs:
ext.include_dirs.append(numpy_incl)
_build_ext.build_extensions(self)
cmdclass = {"build_ext": build_ext}
class BinaryDistribution(Distribution):
def is_pure(self) -> bool:
return False
class CleanCommand(Command):
def run(self) -> None:
raise NotImplementedError("Use git clean -xfd instead")
def initialize_options(self) -> None:
pass
def finalize_options(self) -> None:
pass
cmdclass["clean"] = CleanCommand
with open("README.md") as readme:
description = readme.read()
package_data = defaultdict(list)
filetypes = ["*.csv", "*.csv.gz"]
for root, _, filenames in os.walk(os.path.join(os.getcwd(), "arch")): # noqa: E501
matches = []
for filetype in filetypes:
for filename in fnmatch.filter(filenames, filetype):
matches.append(filename)
if matches:
package_data[".".join(os.path.relpath(root).split(os.path.sep))] = filetypes
package_data["arch"].append("py.typed")
def run_setup(binary: bool = True) -> None:
if not binary:
extensions = []
import logging
logging.warning(
"""
##############################################################################
Building arch WITHOUT compiling the binary. You should ensure that numba is
installed.
##############################################################################
"""
)
else:
directives = {
"language_level": "3",
"cpow": True,
"linetrace": CYTHON_COVERAGE,
"boundscheck": False,
"wraparound": False,
"cdivision": True,
"binding": True,
}
macros = [("NPY_NO_DEPRECATED_API", "NPY_1_7_API_VERSION")]
if CYTHON_COVERAGE:
macros.append(("CYTHON_TRACE", "1"))
ext_modules = []
ext_modules.append(
Extension(
"arch.univariate.recursions",
["./arch/univariate/recursions.pyx"],
define_macros=macros,
)
)
ext_modules.append(
Extension(
"arch.bootstrap._samplers",
["./arch/bootstrap/_samplers.pyx"],
define_macros=macros,
)
)
extensions = cythonize(
ext_modules, force=CYTHON_COVERAGE, compiler_directives=directives
)
setup(
name="arch",
long_description=description,
long_description_content_type="text/markdown",
packages=["arch"] + find_namespace_packages(),
ext_modules=extensions,
package_dir={"arch": "./arch"},
cmdclass=cmdclass,
zip_safe=False,
include_package_data=False,
package_data=package_data,
distclass=BinaryDistribution,
)
try:
build_binary = CYTHON_INSTALLED
build_binary &= os.environ.get("ARCH_NO_BINARY", None) not in ("1", "True", "true")
run_setup(binary=build_binary)
except (
CCompilerError,
ExecError,
PlatformError,
OSError,
ValueError,
):
run_setup(binary=False)
import warnings
warnings.warn(FAILED_COMPILER_WARNING, UserWarning)