causaltune/pyproject.toml at main · py-why/causaltune · 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
[project]
name = "causaltune"
version = "0.3.0"
description = "AutoML for Causal Inference."
authors = [{ name = "Wise" }]
license = "Apache-2.0"
readme = "README.md"
keywords = ["causaltune"]
requires-python = ">=3.10,<3.13"
classifiers = [
    "Programming Language :: Python :: 3 :: Only",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
]
dependencies = [
    # Tightly-coupled / version-sensitive: pinned exactly.
    "dowhy==0.14",
    "econml==0.16.0",
    "FLAML==2.6.0",
    "xgboost==2.1.4",
    "numpy==2.2.6",
    # Unified HPO-backend interface. 0.2.2 is the first release carrying the
    # cross-backend warm-start / resume hooks (optuna enqueue + hyperopt seeded
    # docs) that CausalTune's optuna/hyperopt parity relies on.
    "hiertunehub==0.2.2",
    # Loosely-coupled: floors (with caps where an upstream ceiling is real).
    "pandas>=2,<3",
    "scikit-learn>=1.4,<1.7",
    "category-encoders>=2.6.3",
    # Default HPO backend (numpy-2 clean from 4.0): hard dep.
    "optuna>=4.0",
    "pytest",
    "matplotlib",
    "dcor",
    "holidays",
    "wise-pizza",
    "seaborn",
]

[project.optional-dependencies]
test = [
    "autoflake",
    "black==23.3.0",
    "flake8",
    "isort",
    "pytest-xdist",
    "nbmake",
    "pre-commit",
]
ray = ["ray[tune]>=2.9"]
# Optional HPO backend. hyperopt 0.2.7 imports the deprecated pkg_resources,
# removed in setuptools 81, so the setuptools ceiling ships with this extra.
hyperopt = ["hyperopt>=0.2.7", "setuptools<81"]

[project.urls]
Homepage = "https://github.com/py-why/causaltune"
Repository = "https://github.com/py-why/causaltune"

[tool.poetry]
packages = [{ include = "causaltune" }]
include = ["causaltune/thirdparty/causalml/LICENSE"]

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

[tool.pytest.ini_options]
addopts = "--strict-markers -vv --nbmake --nbmake-timeout=6000"
markers = [
    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
]
testpaths = [
    "tests",
    #"notebooks"
]