forked from bashtage/linearmodels
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
141 lines (120 loc) · 4.66 KB
/
setup.py
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
from setuptools import Extension, find_packages, setup
from setuptools.dist import Distribution
from distutils.errors import CCompilerError, DistutilsExecError, DistutilsPlatformError
import glob
import os
from typing import Dict
try:
from Cython.Build import cythonize
CYTHON_INSTALLED = True
except ImportError:
CYTHON_INSTALLED = False
FAILED_COMPILER_ERROR = """
******************************************************************************
* WARNING *
******************************************************************************
Unable to build binary modules for linearmodels. These are not required
to run any code in the package, and only provide speed-ups for a small subset
of models.
******************************************************************************
* WARNING *
******************************************************************************
"""
with open("README.md") as readme:
long_description = readme.read()
additional_files = ["py.typed"]
for filename in glob.iglob("./linearmodels/datasets/**", recursive=True):
if ".csv.bz" in filename:
additional_files.append(filename.replace("./linearmodels/", ""))
for filename in glob.iglob("./linearmodels/tests/**", recursive=True):
if ".txt" in filename or ".csv" in filename or ".dta" in filename:
additional_files.append(filename.replace("./linearmodels/", ""))
for filename in glob.iglob("./examples/**", recursive=True):
if ".png" in filename:
additional_files.append(filename)
class BinaryDistribution(Distribution):
def is_pure(self) -> bool:
return False
def run_setup(binary: bool = True) -> None:
extensions = []
if binary:
import numpy
macros = [("NPY_NO_DEPRECATED_API", "1")]
# macros.append(('CYTHON_TRACE', '1'))
directives: Dict[str, bool] = {} # {'linetrace': True, 'binding':True}
extension = Extension(
"linearmodels.panel._utility",
["linearmodels/panel/_utility.pyx"],
define_macros=macros,
include_dirs=[numpy.get_include()],
)
extensions.append(extension)
extensions = cythonize(extensions, compiler_directives=directives, force=True)
else:
import logging
logging.warning("Building without binary support")
setup(
name="linearmodels",
license="NCSA",
description="Linear Panel, Instrumental Variable, Asset Pricing, and System "
"Regression models for Python",
packages=find_packages(),
package_dir={"linearmodels": "./linearmodels"},
author="Kevin Sheppard",
author_email="[email protected]",
url="http://github.com/bashtage/linearmodels",
long_description=long_description,
long_description_content_type="text/markdown",
install_requires=open("requirements.txt").read().split("\n"),
include_package_data=True,
package_data={"linearmodels": additional_files},
keywords=[
"linear models",
"regression",
"instrumental variables",
"IV",
"panel",
"fixed effects",
"clustered",
"heteroskedasticity",
"endogeneity",
"instruments",
"statistics",
"statistical inference",
"econometrics",
],
zip_safe=False,
classifiers=[
"Development Status :: 5 - Production/Stable",
"Intended Audience :: End Users/Desktop",
"Intended Audience :: Financial and Insurance Industry",
"Intended Audience :: Science/Research",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"License :: OSI Approved",
"Operating System :: MacOS :: MacOS X",
"Operating System :: Microsoft :: Windows",
"Operating System :: POSIX",
"Programming Language :: Python",
"Topic :: Scientific/Engineering",
],
ext_modules=extensions,
python_requires=">=3.8",
distclass=BinaryDistribution,
)
try:
build_binary = CYTHON_INSTALLED
build_binary &= os.environ.get("LM_NO_BINARY", None) not in ("1", "True", "true")
run_setup(binary=build_binary)
except (
CCompilerError,
DistutilsExecError,
DistutilsPlatformError,
OSError,
ValueError,
ImportError,
):
run_setup(binary=False)
import warnings
warnings.warn(FAILED_COMPILER_ERROR, UserWarning)