pyproject.toml

[project]
name = "opa-database"
version = "1.3.0"
description = "Initial setup for OPA Database."
readme = "README.md"
authors = [{ name = "VictorF13", email = "victorabreuf13@gmail.com" }]
maintainers = [{ name = "VictorF13", email = "victorabreuf13@gmail.com" }]
requires-python = ">=3.12"
license = { text = "MIT" }
dependencies = [
    "click>=8.4.1",
    "pandera>=0.32.0",
    "polars>=1.41.2",
    "psycopg[binary]>=3.3",
    "pydantic>=2.13.4",
    "pydantic-settings>=2.14.1",
]

[project.scripts]
opa-database = "opa_database.cli:cli"

[dependency-groups]
dev = [
    "ipykernel>=7.3.0",
    "jupyter>=1.1.1",
    "prek>=0.4.5",
    "pytest>=9.1.1",
    "ruff>=0.15.18",
    "ty>=0.0.51",
]
ml = [
    "catboost>=1.2.10",
    "folium>=0.20.0",
    "joblib>=1.5.3",
    "lightgbm>=4.7.0",
    "matplotlib>=3.11.1",
    "optuna>=4.9.0",
    "pandas>=3.0.5",
    "scikit-learn>=1.9.0",
    "streamlit>=1.61.1",
    "streamlit-folium>=0.27.4",
    "xgboost>=3.4.1",
]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.ruff]
lint.extend-select = ["ALL"]
lint.ignore = ["PLR0913", "D203", "D213", "COM812"]

[tool.ruff.lint.per-file-ignores]
"tests/**" = ["S101", "SLF001", "ANN201", "D103"]
"*.ipynb" = ["T201"]
# Progress-reporting CLI scripts: stdout is the interface, and they are
# run directly (uv run path/to/script.py) rather than imported, so they
# are deliberately not packages.
"ml/*/scripts/*.py" = ["T201", "INP001"]

[tool.ty]
src.include = ["src", "tests", "ml/trip_validity_model/app"]
# trip_validity_model/app listed first: it and bus_matching_model/app have
# same-named sibling modules (features.py, db.py, ...), so with both paths
# present, `from features import X` resolves against whichever is listed
# first for any file ty checks -- trip_validity's own files must resolve
# to themselves. bus_matching_model/app is here only so the notebooks
# under ml/bus_matching_model/notebooks/ can resolve `gtfs_cache` (a name
# that doesn't collide); bus_matching_model/app's own files are excluded
# from the ty pre-commit hook entirely (see .pre-commit-config.yaml) so
# they never hit the collision themselves.
environment.extra-paths = ["ml/trip_validity_model/app", "ml/bus_matching_model/app"]
rules.division-by-zero = "warn"
rules.possibly-unresolved-reference = "warn"
rules.unused-ignore-comment = "warn"

[tool.semantic_release]
version_toml = ["pyproject.toml:project.version"]
commit_message = "chore(release): v{version}"
commit_author = "Victor Abreu <victorabreuf13@gmail.com>"
remote.type = "github"
changelog.default_templates.changelog_file = "CHANGELOG.md"
build_command = """
uv lock
git add uv.lock
"""