metalearners/pyproject.toml at main · Quantco/metalearners · 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
106
107
[build-system]
requires = ["hatchling", "hatch-vcs"]
build-backend = "hatchling.build"

[project]
name = "metalearners"
description = "MetaLearners for CATE estimation"
dynamic = ["version"]
authors = [
  {name = "QuantCo, Inc.", email = "noreply@quantco.com"},
]
classifiers = [
  "License :: OSI Approved :: BSD License",
  "Operating System :: OS Independent",
  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3.10",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: 3.13",
]
readme = "README.md"
requires-python = ">=3.10"
dependencies = [
    "scikit-learn>=1.4,<1.8",
    "pandas",
    "numpy",
    "narwhals>=1.31.0",
    "typing-extensions",
    "shap",
    "joblib>=1.2.0"
]

[project.urls]
repository = "https://github.com/quantco/metalearners"

[tool.hatch.version]
source = "vcs"

[tool.hatch.build.targets.wheel]
packages = ["metalearners"]

[project.scripts]

[tool.docformatter]
black = true

[tool.black]
exclude = '''
/(
    \.eggs
  | \.git
  | \.venv
  | build
  | dist
)/
'''

[tool.ruff]
line-length = 88

[tool.ruff.lint]
ignore = [
  "N803",  # https://docs.astral.sh/ruff/rules/invalid-argument-name
  "N806",  # https://docs.astral.sh/ruff/rules/non-lowercase-variable-in-function
  "E501",  # https://docs.astral.sh/ruff/faq/#is-the-ruff-linter-compatible-with-black
]
select = [
  # pyflakes
  "F",
  # pycodestyle
  "E", "W",
  # isort
  "I",
  # pep8-naming
  "N",
  # pyupgrade
  "UP",
]

[tool.mypy]
python_version = '3.10'
ignore_missing_imports = true
no_implicit_optional = true
check_untyped_defs = true

exclude = ["docs/"]

[tool.pytest.ini_options]
addopts = "--import-mode=importlib --cov=metalearners --cov-report term-missing --color=yes"
testpaths = [
  "tests",
]

[tool.typos.default]
extend-ignore-re = [
  "HTE",
  "crace",
  "frst_in_family",
  # Disable spell-checking for an individual line.
  "(?Rm)^.*(#|//)\\s*spellchecker:disable-line$",
  # Disable spell-checking for a code block.
  "(?s)(#|//)\\s*spellchecker:off.*?\\n\\s*(#|//)\\s*spellchecker:on",
]

[tool.typos.default.extend-words]
# statsmodels' standard errors attribute, e.g. `RegressionResults.bse`.
bse = "bse"