|
1 | | -import os |
2 | | - |
3 | 1 | import setuptools |
4 | 2 |
|
5 | | -here = os.path.abspath(os.path.dirname(__file__)) |
6 | | - |
7 | | -with open("README.md", encoding="UTF-8") as fh: |
8 | | - long_description = fh.read() |
9 | | - |
10 | | - |
11 | | -# Get the code version |
12 | | -version = {} |
13 | | -with open(os.path.join(here, "flaml/version.py")) as fp: |
14 | | - exec(fp.read(), version) |
15 | | -__version__ = version["__version__"] |
16 | | - |
17 | | -install_requires = [ |
18 | | - "NumPy>=1.17", |
19 | | -] |
20 | | - |
21 | | - |
22 | | -setuptools.setup( |
23 | | - name="FLAML", |
24 | | - version=__version__, |
25 | | - author="Microsoft Corporation", |
26 | | - author_email="hpo@microsoft.com", |
27 | | - description="A fast library for automated machine learning and tuning", |
28 | | - long_description=long_description, |
29 | | - long_description_content_type="text/markdown", |
30 | | - url="https://github.com/microsoft/FLAML", |
31 | | - packages=setuptools.find_packages(include=["flaml*"]), |
32 | | - package_data={ |
33 | | - "flaml.default": ["*/*.json"], |
34 | | - }, |
35 | | - include_package_data=True, |
36 | | - install_requires=install_requires, |
37 | | - extras_require={ |
38 | | - "automl": [ |
39 | | - "lightgbm>=2.3.1", |
40 | | - "xgboost>=0.90,<3.0.0", |
41 | | - "scipy>=1.4.1", |
42 | | - "pandas>=1.1.4", |
43 | | - "scikit-learn>=1.0.0", |
44 | | - ], |
45 | | - "notebook": [ |
46 | | - "jupyter", |
47 | | - ], |
48 | | - "spark": [ |
49 | | - "pyspark>=3.2.0", |
50 | | - "pandas<3", |
51 | | - "joblibspark>=0.5.0", |
52 | | - "joblib<=1.3.2", |
53 | | - ], |
54 | | - "test": [ |
55 | | - "numpy>=1.17,<2.0.0; python_version<'3.13'", |
56 | | - "numpy>=1.17; python_version>='3.13'", |
57 | | - "jupyter", |
58 | | - "lightgbm>=2.3.1", |
59 | | - "xgboost>=0.90,<2.0.0; python_version<'3.11'", |
60 | | - "xgboost>=2.0.0; python_version>='3.11'", |
61 | | - "scipy>=1.4.1", |
62 | | - "pandas>=1.1.4,<2.0.0; python_version<'3.10'", |
63 | | - "pandas>=1.1.4; python_version>='3.10'", |
64 | | - "scikit-learn>=1.2.0", |
65 | | - "thop", |
66 | | - "pytest>=6.1.1", |
67 | | - "pytest-rerunfailures>=13.0", |
68 | | - "coverage>=5.3", |
69 | | - "pre-commit", |
70 | | - "torch", |
71 | | - "torchvision", |
72 | | - "catboost>=0.26", |
73 | | - "rgf-python", |
74 | | - "optuna>=2.8.0,<=3.6.1", |
75 | | - "openml", |
76 | | - "statsmodels>=0.12.2", |
77 | | - "psutil", |
78 | | - "dataclasses", |
79 | | - "transformers[torch]", |
80 | | - "datasets", |
81 | | - "evaluate", |
82 | | - "nltk!=3.8.2", # 3.8.2 doesn't work with mlflow |
83 | | - "rouge_score", |
84 | | - "hcrystalball", |
85 | | - "seqeval", |
86 | | - "pytorch-forecasting", |
87 | | - "mlflow-skinny<=2.22.1", # Refer to https://mvnrepository.com/artifact/org.mlflow/mlflow-spark |
88 | | - "joblibspark>=0.5.0", |
89 | | - "joblib<=1.3.2", |
90 | | - "nbconvert", |
91 | | - "nbformat", |
92 | | - "ipykernel", |
93 | | - "pytorch-lightning", # test_forecast_panel |
94 | | - "tensorboardX", # test_forecast_panel |
95 | | - "requests", # https://github.com/docker/docker-py/issues/3113 |
96 | | - "packaging", |
97 | | - "dill", # a drop in replacement of pickle |
98 | | - ], |
99 | | - "catboost": [ |
100 | | - "catboost>=0.26", |
101 | | - ], |
102 | | - "blendsearch": [ |
103 | | - "optuna>=2.8.0,<=3.6.1", |
104 | | - "packaging", |
105 | | - ], |
106 | | - "ray": [ |
107 | | - "ray[tune]>=1.13,<2.5.0", |
108 | | - ], |
109 | | - "azureml": [ |
110 | | - "azureml-mlflow", |
111 | | - ], |
112 | | - "nni": [ |
113 | | - "nni", |
114 | | - ], |
115 | | - "vw": [ |
116 | | - "vowpalwabbit>=8.10.0, <9.0.0", |
117 | | - "scikit-learn", |
118 | | - ], |
119 | | - "hf": [ |
120 | | - "transformers[torch]>=4.26", |
121 | | - "datasets", |
122 | | - "nltk<=3.8.1", |
123 | | - "rouge_score", |
124 | | - "seqeval", |
125 | | - ], |
126 | | - "nlp": [ # for backward compatibility; hf is the new option name |
127 | | - "transformers[torch]>=4.26", |
128 | | - "datasets", |
129 | | - "nltk<=3.8.1", |
130 | | - "rouge_score", |
131 | | - "seqeval", |
132 | | - ], |
133 | | - "ts_forecast": [ |
134 | | - "holidays", |
135 | | - "prophet>=1.1.5", |
136 | | - "statsmodels>=0.12.2", |
137 | | - "hcrystalball>=0.1.10", |
138 | | - ], |
139 | | - "forecast": [ |
140 | | - "holidays", |
141 | | - "prophet>=1.1.5", |
142 | | - "statsmodels>=0.12.2", |
143 | | - "hcrystalball>=0.1.10", |
144 | | - "pytorch-forecasting>=0.10.4", |
145 | | - "pytorch-lightning>=1.9.0", |
146 | | - "tensorboardX>=2.6", |
147 | | - ], |
148 | | - "benchmark": ["catboost>=0.26", "psutil==5.8.0", "xgboost==1.3.3", "pandas==1.1.4"], |
149 | | - "synapse": [ |
150 | | - "joblibspark>=0.5.0", |
151 | | - "optuna>=2.8.0,<=3.6.1", |
152 | | - "pyspark>=3.2.0", |
153 | | - ], |
154 | | - "autozero": ["scikit-learn", "pandas", "packaging"], |
155 | | - }, |
156 | | - classifiers=[ |
157 | | - "License :: OSI Approved :: MIT License", |
158 | | - "Operating System :: OS Independent", |
159 | | - # Specify the Python versions you support here. |
160 | | - "Programming Language :: Python :: 3", |
161 | | - "Programming Language :: Python :: 3.10", |
162 | | - "Programming Language :: Python :: 3.11", |
163 | | - "Programming Language :: Python :: 3.12", |
164 | | - ], |
165 | | - python_requires=">=3.10", |
166 | | -) |
| 3 | +setuptools.setup() |
0 commit comments