pyproject.toml

[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "databricks-impulse"
dynamic = ["version"]
description = "Large-scale time-series measurement data analytics on Apache Spark"
readme = "README.md"
requires-python = ">=3.12,<3.13"
license = {file = "LICENSE.txt"}
authors = [
    {name = "Databricks Labs"}
]
maintainers = [
    {name = "Databricks Labs"}
]
dependencies = [
    "lz4==4.4.5",
]

# Pre-installed on Databricks Serverless env v4 (and DBR ML runtimes).
# Declared as opt-in extras so the wheel installs cleanly on Databricks
# without conflicting with pre-installed runtime versions. Local dev and
# CI install these via `uv sync --all-extras`.
[project.optional-dependencies]
local-dev = [
    "databricks-sdk==0.106.0",
    "matplotlib==3.10.0",
    "numpy==2.1.3",
    "pandas==2.2.3",
    "pyarrow==19.0.1",
    "pydantic==2.11.7",
    "pyspark==4.0.0",
    "delta-spark==4.0.1",
    "scipy==1.15.1",
]

[project.urls]
Homepage = "https://github.com/databrickslabs/impulse"
Repository = "https://github.com/databrickslabs/impulse"
Documentation = "https://databrickslabs.github.io/impulse"
Issues = "https://github.com/databrickslabs/impulse/issues"

[dependency-groups]
dev = [
    "pre-commit==4.1.0",
    "pydoc-markdown==4.8.2",
    "pytest==8.3.5",
    "pytest-benchmark==5.1.0",
    "pytest-cov==6.0.0",
    "pytest-mock==3.15.1",
    "pytest-xdist==3.6.1",
    "ruff==0.9.10",
    "setuptools==75.8.2",
    "black==26.3.1"
]

# Setuptools options
[tool.setuptools.dynamic]
version = {file = ["VERSION"]}

[tool.setuptools.packages.find]
where = ["src"]
exclude = ["src/*/demos"]

# Pytest options
[tool.pytest.ini_options]
pythonpath = [".", "src/impulse_reporting"]
testpaths = ["tests"]
addopts = "--junitxml=test-results.xml --cov=src --cov-report=html --cov-report=xml:coverage.xml --cov-report=term-missing"

# Default dependency groups
[tool.uv]
required-version = "~= 0.11.0"
default-groups = ["dev"]

# Coverage options
[tool.coverage.run]
branch = true
source = ["src"]

[tool.coverage.report]
omit = ["*/__init__.py"]

[tool.ruff]
line-length = 99
exclude = [".git", ".venv", "venv", "__pycache__", "build", "dist"]

[tool.ruff.lint]
select = [
    # pycodestyle
    "E",
    # Pyflakes
    "F",
    # flake8-bugbear
    "B",
    # isort
    "I",
    # flake8-simplify
    "SIM",
    # flake8-unused-arguments
    "ARG",
    # pyupgrade — modern type hints
    "UP006",   # List[int] → list[int]
    "UP007",   # Union[X, Y] → X | Y, Optional[X] → X | None
    "UP035",   # deprecated typing imports → collections.abc
]
ignore = [
    "E501",
    "E712",
    "E722",
    "B008",
    "F811",
    "I001",   # Import block is un-sorted or un-formatted (isort)
    "SIM118", # Use 'key in dict' instead of 'key in dict.keys()'
    "SIM108", # Use ternary operator instead of if-else-block
    "SIM114", # Combine if branches using logical 'or'
    "SIM105", # Use contextlib.suppress instead of try-except-pass
    "SIM300", # Yoda conditions (e.g., '1 == len(x)')
    "ARG001", # Unused function argument
    "ARG002", # Unused method argument
    "ARG003", # Unused class method argument
    "F821",   # Undefined name (use with caution)
    "F841",   # Local variable assigned but never used
    "F401",   # Unused imports
    "B023",   # Function definition does not bind loop variable
    "B027",   # Empty method in ABC without abstract decorator
    "E402",   # Module level import not at top of file
    "E731",   # Do not assign a lambda expression
    "E711",   # Comparison to None (should use 'is None')
]

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

[tool.ruff.lint.isort]
known-first-party = ["impulse_query_engine", "impulse_reporting"]

[tool.black]
line-length = 99
target-version = ['py312']
include = '\.pyi?$'
extend-exclude = '''
/(
  # directories
  \.eggs
  | \.git
  | \.hg
  | \.mypy_cache
  | \.tox
  | \.venv
  | venv
  | _build
  | buck-out
  | build
  | dist
)/
'''