pyproject.toml

[project]
name = "tusk-ml"
version = "0.1.7"
description = "Automated feature engineering for narwhals dataframes"
authors = [
    {name="Excidion", email="excidion@proton.me"}
]
readme = "README.md"
license = "Apache-2.0"
requires-python = ">=3.10"
dependencies = ["narwhals>=2.24"]

[project.urls]
"Documentation" = "https://excidion.github.io/tusk/"
"Source Code"= "https://github.com/Excidion/tusk"

[project.optional-dependencies]
sklearn = ["scikit-learn>=1.4"]
plot = ["mermaidx>=0.9.5"]

[build-system]
requires = ["uv_build>=0.12.7,<0.13"]
build-backend = "uv_build"

[tool.uv.build-backend]
# The distribution is named tusk-ml because tusk is taken on PyPI; the
# import name stays tusk.
module-name = "tusk"


[dependency-groups]
lint = [
    "interrogate>=1.7.0",
    "pre-commit-uv>=4.3.0",
    "pydoclint>=0.9.1",
    "ruff>=0.16.3",
    "ty>=0.0.72",
]
test = [
    "duckdb>=1.0",
    "mermaidx>=0.9.5",
    "pandas>=2.0",
    "polars>=1.43",
    "pyarrow>=17",
    "pytest>=8.0",
    "scikit-learn>=1.4",
]
dev = [
    { include-group = "lint" },
    { include-group = "test" },
]
validation = [
    "featuretools>=1.31",
    # featuretools imports pkg_resources, which setuptools 81 removed.
    "setuptools<81",
]
benchmark = [
    "relbench>=2.1",
]
docs = [
    "mkdocstrings[python]>=1.0.6",
    "zensical>=0.0.56",
]

[tool.pytest.ini_options]
testpaths = ["tests", "benchmarks"]
markers = [
    "differential: cross-checks values against featuretools (opt-in)",
    "benchmark: relbench performance runs (opt-in)",
]
addopts = "-m 'not differential and not benchmark'"

[tool.ruff.lint]
select = ["E", "F", "I", "UP", "B", "D"]

[tool.ruff.lint.pydocstyle]
convention = "google"

[tool.ruff.lint.per-file-ignores]
"tests/*" = ["D"]

[tool.interrogate]
fail-under = 100
ignore-init-module = true
ignore-private = true
ignore-magic = true
ignore-nested-functions = true
exclude = ["tests", ".venv", ".claude"]

[tool.pydoclint]
style = "google"
exclude = "tests|benchmarks"
arg-type-hints-in-docstring = false
check-return-types = false
# ruff (D107) and interrogate both require __init__ to carry its own
# docstring; without this flag pydoclint's DOC301 forbids exactly that
# whenever the class also has a docstring.
allow-init-docstring = true

[[tool.ty.overrides]]
include = ["src/tusk/primitives/**", "tests/**"]

[tool.ty.overrides.rules]
# Primitive.build is variadic in the ABC and fixed-arity in each concrete
# primitive. Arity is guaranteed by construction: synthesis supplies exactly
# len(input_dtypes) expressions to outputs(), which is build()'s only caller.
# Scoped rather than project-wide so override checking stays active everywhere
# else.
invalid-method-override = "ignore"