XGBoostLSS/pyproject.toml at master · StatMixedML/XGBoostLSS · GitHub

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
[project]
name = "xgboostlss"
version = "0.6.1"
description = "XGBoostLSS - An extension of XGBoost to probabilistic modelling"
readme = { file = "README.md", content-type = "text/markdown" }
authors = [
    { name = "Alexander März", email = "statmixedmlgit@gmail.com" }
]
license = { text = "Apache License 2.0" }
requires-python = ">=3.11"
classifiers = [
  "Intended Audience :: Developers",
  "Intended Audience :: Science/Research",
  "License :: OSI Approved :: Apache Software License",
  "Operating System :: MacOS",
  "Operating System :: Microsoft :: Windows",
  "Operating System :: POSIX",
  "Operating System :: Unix",
  "Programming Language :: Python",
  "Programming Language :: Python :: 3 :: Only",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: 3.13",
  "Programming Language :: Python :: 3.14",
  "Topic :: Scientific/Engineering",
  "Topic :: Scientific/Engineering :: Artificial Intelligence",
  "Topic :: Software Development",
]
# core dependencies of xgboostlss
# this set should be kept minimal!
dependencies = [
    "xgboost>=2.0,<4.0",
    "torch>=2.1,<2.10",
    "pyro-ppl>=1.8,<1.10",
    "numpy>=1.26.0,<2.4",
    "pandas>=2.0,<2.3",
    "scipy>=1.11,<1.17",
    "tqdm>=4.66,<4.68",
    "matplotlib>=3.8,<3.11",
    "scikit-base",
]

[project.optional-dependencies]
all_extras = [
    "optuna>=3.5.0,<4.4",
    "optuna-integration",
    "seaborn>=0.13.0,<0.14.0",
    'shap>=0.50.0; python_version < "3.14"',
]
docs = [
    "mkdocs",
    "mkdocstrings[python]",
    "mkdocs-jupyter"
]
notebooks = [
    "scikit-learn>=1.4.0,<1.8.0",
    "plotnine>=0.12.0,<0.16.0",
]
test = [
    "flake8",
    "pytest",
    "pytest-coverage",
    "pytest-timeout",
]

[project.urls]
Homepage = "https://github.com/StatMixedML/XGBoostLSS"

[tool.setuptools]
include-package-data = true
zip-safe = true

[tool.setuptools.packages.find]
exclude = ["docs", "tests*"]

[tool.setuptools.package-data]
xgboostlss = ["datasets/*.csv"]

[build-system]
requires = ["setuptools>=61", "wheel"]
build-backend = "setuptools.build_meta"