pybandits/pyproject.toml at develop · PlaytikaOSS/pybandits · 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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
[tool.poetry]
name = "pybandits"
version = "6.0.4"
description = "Python Multi-Armed Bandit Library"
authors = [
    "Dario d'Andrea <dariod@playtika.com>",
    "Shahar Bar <shaharbar1@gmail.com>",
    "Jerome Carayol <jeromec@playtika.com>",
    "Anastasiia Kabeshova <anastasiia.kabeshova@gmail.com>",
    "Stefano Piazza <stefanop@playtika.com>",
    "Ron Shiff <ron.shiff1@gmail.com>",
    "Raphael Steinmann <raphaels@playtika.com>",
    "Yoav Tulpan <yoavtp@gmail.com>",
    "Armand Valsesia <armandv@playtika.com>",
]
license = "MIT License"
readme = "README.md"
classifiers = [
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
]
homepage = "https://github.com/PlaytikaOSS/pybandits"
repository = "https://github.com/PlaytikaOSS/pybandits"
keywords = ["multi-armed bandits", "reinforcement-learning", "optimization"]

[tool.poetry.dependencies]
python = ">=3.9,<3.13"
loguru = "^0.6"
numpy = [
    { version = ">=1.25", python = ">=3.9,<3.12" },
    { version = ">=1.26", python = "3.12.*" },
]
pydantic = ">=1.10,<3"
scipy = [
    { version = "^1.9", python = ">=3.9,<3.12" },
    { version = "^1.11", python = "3.12.*" },
]
numpyro = [
    { version = "^0.16", python = ">=3.9,<3.11" },
    { version = "^0.20", python = ">=3.11" },
]
scikit-learn = "^1.1"
optuna = "^3.6"
tqdm = "^4"
bokeh = "^3.1"
xgboost = {version = "^2", optional=true}
ipython = {version = "^8", optional=true}
jax = "<=0.9.2"

[tool.poetry.group.dev.dependencies]
hypothesis = "^6.68.2"
pytest = "^8.3.3"
tox = "^4.4.7"
pandas = ">=1.5.3"
pre-commit = ">=3.1.1"
nbstripout = "^0.8.1"
ipykernel = "^6.21.3"
jupyterlab = "^3.6.1"
pytest-cov = "^5.0.0"
pytest_mock = "^3.14.0"
pytest-xdist = "^3.6.1"
ruff = "^0.11.0"
sphinx = "^7.1"
sphinx_rtd_theme = "^2.0"
nbsphinx = "^0.9"
myst-parser = "^3.0"
pandoc = "^2.4"
matplotlib = "^3.5"


[tool.poetry.extras]
ope = ["xgboost"]
jupyter = ["ipython"]


[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"

[tool.mypy]
plugins = "pydantic.mypy"

[tool.ruff]
line-length = 120
extend-include = ["*.ipynb"]


# pylint configuration incorporated in Ruff
[tool.ruff.lint]
# Enable the isort rules.
extend-select = ["I"]
extend-ignore = ["E203"]
# disable check for:
# D100: Missing docstring in public module (equivalent to C0114)
# D101: Missing docstring in public class (equivalent to C0115)
# D102: Missing docstring in public method (equivalent to C0116)
# D103: Missing docstring in public function (equivalent to C0116)
# D104: Missing docstring in public package (equivalent to C0114)
# D105: Missing docstring in magic method (equivalent to C0116)
# D106: Missing docstring in public nested class (equivalent to C0115)
# Missing : Too few public methods (equivalent to R0903)
# PLR0913: Too many arguments (equivalent to R0913)
per-file-ignores = { "*.py" = ["D100", "D101", "D102", "D103", "D104", "D105", "D106", "PLR0913"] }