-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
30 lines (29 loc) · 879 Bytes
/
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
from setuptools import find_packages, setup
setup(
name="pbmohpo",
version="0.1",
description="Preferential Bayesian Multi-objective Hyperparameter Optimization",
url="https://github.com/ja-thomas/pbmohpo",
author="Janek Thomas",
author_email="[email protected]",
license="LGPL",
packages=find_packages(),
install_requires=[
"ConfigSpace==0.6.0",
"botorch==0.9.2",
"gpytorch",
"torch",
"yahpo-gym @ git+https://github.com/slds-lmu/yahpo_gym.git@v2#subdirectory=yahpo_gym",
"onnxruntime<1.16.0",
"openml",
"lightgbm",
"matplotlib",
"numpy",
],
extras_require={
"dev": ["black", "flake8", "isort"],
"experiments": ["yacs", "mlflow"],
"test": ["pytest>=4.6"],
"docs": ["sphinx", "sphinx_rtd_theme"],
},
)