Skip to content

Commit

Permalink
--wip-- [skip ci]
Browse files Browse the repository at this point in the history
  • Loading branch information
thomasmarwitz committed Nov 26, 2024
1 parent 8f86675 commit 2306f17
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 1 deletion.
1 change: 0 additions & 1 deletion metalearners/rlearner.py
Original file line number Diff line number Diff line change
Expand Up @@ -346,7 +346,6 @@ def predict(
tau_hat[variant_indices, treatment_variant - 1] = variant_estimates
return tau_hat

@copydoc(MetaLearner.evaluate, sep="\n\t")
def evaluate(
self,
X: Matrix,
Expand Down
10 changes: 10 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,14 @@ 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
"D100",
"D101",
"D102",
"D103",
"D104",
"D105",
"D106",
"D107",
]
select = [
# pyflakes
Expand All @@ -73,6 +81,8 @@ select = [
"N",
# pyupgrade
"UP",
# pydocstyle
"D",
]

[tool.mypy]
Expand Down

0 comments on commit 2306f17

Please sign in to comment.