ml/python/pyproject.toml at main · systats/ml · 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
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "mlw"
version = "1.2.0"
description = "ML that works. One import, the Hastie workflow in code."
readme = "README.md"
license = "MIT"
requires-python = ">=3.10"
authors = [{name = "Simon Roth"}]
keywords = ["machine-learning", "ml", "scikit-learn", "xgboost", "classification", "regression"]
classifiers = [
    "Development Status :: 5 - Production/Stable",
    "Intended Audience :: Science/Research",
    "Intended Audience :: Education",
    "License :: OSI Approved :: MIT License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3.13",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Education",
    "Typing :: Typed",
]
dependencies = [
    "pandas>=2.0,<3.0",
    "numpy>=1.24,<3.0",
    "skops>=0.9,<2.0",
    "ml-py>=1.0.0",
]

[project.optional-dependencies]
sklearn = ["scikit-learn>=1.4,<2.0"]
scipy = ["scipy>=1.10"]
xgboost = ["xgboost>=2.0"]
lightgbm = ["lightgbm>=4.0"]
catboost = ["catboost>=1.2"]
optuna = ["optuna>=3.0"]
shap = ["shap>=0.42"]
tabpfn = ["tabpfn>=0.1,<2.0"]
plots = ["matplotlib>=3.5"]
polars = ["narwhals>=1.0"]
all = [
    "scikit-learn>=1.4,<2.0", "scipy>=1.10",
    "xgboost>=2.0", "lightgbm>=4.0", "catboost>=1.2",
    "optuna>=3.0", "shap>=0.42", "narwhals>=1.0", "matplotlib>=3.5",
]
dev = [
    "pytest>=7.0", "pytest-cov", "pytest-timeout>=2.0", "ruff",
    "hypothesis>=6.0",
    "scikit-learn>=1.4,<2.0", "scipy>=1.10",
    "xgboost>=2.0", "lightgbm>=4.0", "optuna>=3.0",
]

[project.urls]
Repository = "https://github.com/epagogy/ml"
Documentation = "https://epagogy.ai"
Changelog = "https://github.com/epagogy/ml/blob/main/CHANGELOG.md"

[tool.hatch.build.targets.sdist]
exclude = [
    "catboost_info/",
    "dist/",
    "*.pyml",
    "*.ml",
    "*.html",
    "FRICTION_REPORT_DR_PATEL.md",
    ".hypothesis/",
    "notebooks/",
    "tests/",
    "backtest.ipynb",
    "quickstart.ipynb",
]

[tool.hatch.build.targets.wheel]
packages = ["ml"]
artifacts = ["ml/py.typed"]

[tool.pytest.ini_options]
testpaths = ["tests"]
addopts = "--tb=short -m 'not slow' --timeout=60"
markers = [
    "smoke: critical path, <3s total",
    "slow: dataset-heavy or long-running, excluded from default run",
]
filterwarnings = [
    "ignore::DeprecationWarning:matplotlib",
    "ignore::DeprecationWarning:fontTools",
]
norecursedirs = [".git", ".hypothesis", "__pycache__", "*.egg-info", "catboost_info"]

[tool.ruff]
line-length = 100
target-version = "py310"

[tool.ruff.lint]
select = ["E", "F", "W", "I", "UP", "B", "SIM"]
ignore = ["E501", "SIM102", "SIM108", "SIM116", "B905"]

[tool.ruff.lint.per-file-ignores]
"__init__.py" = ["I001"]