pyproject.toml

[project]
name = "mwl-finetuning"
version = "1.0.0"
description = "More-With-Less Fine-Tuning Library"
readme = "README.md"
authors = [{ name = "Merantix Momentum GmbH" }]
requires-python = ">=3.11"
dependencies = [
    "datasets>=4.4.1",
    "hydra-zen>=0.16.0",
    "kaleido>=1.2.0",
    "lm-eval>=0.4.9.2",
    "openai>=1.0,<2",
    "pandas>=2.3.3,<3.0",
    "peft>=0.15.0",
    "pip>=25.3",
    "plotly>=6.5.0",
    "pre-commit>=4.5.0",
    "rootutils>=1.0.7",
    "sentencepiece>=0.2.1",
    "torch>=2.9.1",
    "transformers>=4.57.3",
    "wandb>=0.19.0",
]

[project.optional-dependencies]
demo = [
    "gradio>=5.0.0",
]
vllm = [
    "vllm>=0.6.0",
]
all = [
    "mwl-finetuning[demo,vllm]",
]

[tool.black]
line-length = 120
include = '\.pyi?$'

[tool.isort]
case_sensitive = false
line_length = 120
order_by_type = false
profile = "black"

[tool.ruff]
line-length = 120
target-version = "py310"
ignore = [
    # E203: Space before :. Deactivated because of incompatibility with black, see https://github.com/ambv/black#slices.
    "E203",
    # E402: Module level import not at top of file. We sometimes need this (e.g. mparams).
    "E402",
    "W605",
    # Exceptions for the type hinting plugin, flake8-annotations
    "ANN002", # Missing type annotation for *args
    "ANN003", # Missing type annotation for **kwargs
    "ANN101", # Missing type annotation for self in method
    "ANN102", # Missing type annotation for cls in classmethod
    "ANN204", # Missing return type annotation for special method, e.g. init
    # The following is for the docstring plugin, to make it less whiny. We are happy if we have docs on all functions
    "D100", # Missing docstring in public module
    "D101", # Missing docstring in public class
    "D104", # Missing docstring in public package
    "D202", # No blank lines allowed after function docstring
    "D205", # 1 blank line required between summary line and description
    "D212", # Multi-line docstring summary should start at the first line
    "D415", # First line should end with a period, question mark, or exclamation point
]

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

[tool.mypy]
explicit_package_bases = true
ignore_missing_imports = true
implicit_optional = true
allow_redefinition = true

[tool.pytest.ini_options]
testpaths = ["tests"]
addopts = "-p no:warnings -v"
norecursedirs = [".*", "build", "dist", "CVS", "_darcs", "{arch}", "*.egg"]

[tool.coverage.report]
exclude_lines = [
    # if a line gets matched with this, we dont count it as missing in the coverage report
    # keep default behaviour of coverage
    "pragma: no cover",
    # skip lines that raise a specific error (^\s matches beginning of line + any whitespace length)
    "^\\s*raise NotImplementedError",
    "^\\s*raise AssertionError",
    "^\\s*raise ValueError",
    # skip lines that are only "pass"
    "^\\s*pass\\s*$",
]