|
1 | | -[build-system] |
2 | | -requires = ["setuptools", "wheel"] |
3 | | -build-backend = "setuptools.build_meta" |
| 1 | +[tool.poetry] |
| 2 | +name = "tpot" |
| 3 | +description = "Tree-based Pipilene Optimization Tool" |
| 4 | +license = "GNU/LGPLv3" # TODO: update this |
| 5 | +version="1.0.0" # TODO: update this |
| 6 | +authors = [ # TODO: this doesn't seem to match code comments |
| 7 | + "Pedro Ribioro", |
| 8 | +] |
| 9 | +readme = "README.md" |
| 10 | +homepage = "https://epistasislab.github.io/tpot/" |
| 11 | +repository="https://github.com/EpistasisLab/tpot2" |
| 12 | +classifiers = [ |
| 13 | + "Intended Audience :: Science/Research", |
| 14 | + "License :: OSI Approved :: GNU Lesser General Public License v3 (LGPLv3)", |
| 15 | + "Programming Language :: Python :: 3.10", |
| 16 | + "Topic :: Scientific/Engineering :: Artificial Intelligence" |
| 17 | +] |
| 18 | +keywords = [ |
| 19 | + "pipeline optimization", |
| 20 | + "hyperparameter optimization", |
| 21 | + "data science", |
| 22 | + "machine learning", |
| 23 | + "genetic programming", |
| 24 | + "evolutionary computation" |
| 25 | +] |
| 26 | + |
| 27 | +[tool.poetry.dependencies] |
| 28 | +ConfigSpace = "*" |
| 29 | +dask = "*" |
| 30 | +joblib = "*" |
| 31 | +lightgbm = "*" |
| 32 | +networkx = "*" |
| 33 | +nose = "*" |
| 34 | +numpy = "*" |
| 35 | +pandas = "*" |
| 36 | +python = "~3.12.0" |
| 37 | +scikit-learn = "*" |
| 38 | +scipy = "*" |
| 39 | +stopit = "*" |
| 40 | +tqdm = "*" |
| 41 | +traitlets = "*" |
| 42 | + |
| 43 | +[tool.poetry.group.cpu] |
| 44 | +optional = true |
| 45 | +[tool.poetry.group.cpu.dependencies] |
| 46 | +xgboost-cpu = "*" |
| 47 | + |
| 48 | +[tool.poetry.group.gpu] |
| 49 | +optional = true |
| 50 | +[tool.poetry.group.gpu.dependencies] |
| 51 | +xgboost = "*" |
| 52 | + |
| 53 | +[tool.poetry.group.graph] |
| 54 | +optional = true |
| 55 | +[tool.poetry.group.graph.dependencies] |
| 56 | +matplotlib = "*" |
| 57 | +networkx = "*" |
| 58 | + |
| 59 | +[tool.poetry.group.nn] |
| 60 | +optional = true |
| 61 | +[tool.poetry.group.nn.dependencies] |
| 62 | +pytorch = "*" |
| 63 | + |
| 64 | +[tool.poetry.group.dev] |
| 65 | +optional = true |
| 66 | +[tool.poetry.group.dev.dependencies] |
| 67 | +nose = "*" |
| 68 | +pytest = "*" |
| 69 | +pytest-cov = "*" |
| 70 | +mypy = "*" |
| 71 | +flake8 = "*" |
| 72 | +tox = "*" |
| 73 | + |
4 | 74 |
|
5 | 75 | [tool.pytest.ini_options] |
6 | 76 | addopts = "--cov=tpot" |
|
0 commit comments