pyproject.toml

# Copyright 2026 Kaggle Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

[project]
name = "kaggle_benchmarks"
dynamic=["version"]
description = ""
readme = "README.md"
authors = []
requires-python = ">=3.11"
# NOTE: Before upgrading dependencies, bump [tool.uv] exclude-newer date below.
dependencies = [
  "openai>=1.66",
  "pandas",
  "tqdm",
  "python-dotenv>=1.0.1",
  "protobuf>=5.29.3,<6.0",
  "jupyter",
  "docker>=7.1.0",
  "panel>=1.6.3",
  "playwright>=1.50.0",
  "nest-asyncio>=1.6.0",
  "hishel[httpx]>1.1.8",
  "pydantic>=2.11.5",
  "joblib",
  "google-genai>=1.38.0",
]

[project.optional-dependencies]
kaggle-client = [
  "jupytext",
]

# Configuration for the build backend to find the version file
[tool.hatch.version]
path = "src/kaggle_benchmarks/__init__.py"

[tool.hatch.build.targets.sdist]
exclude = ["skills/"]

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

[dependency-groups]
test = [
    "pytest",
    "pytest-asyncio>=0.25.3",
    "pytest-mock>=3.14.1",
    "pytest-xdist>=3.8.0",
    "respx>=0.22.0",
    "kaggle_benchmarks[kaggle-client]",
]
kaggle = [
  "kagglehub>=0.3.10",
  "pip>=25.0.1",
  "nbdev>=2.4.2",
  "papermill>=2.6.0",
  # b/276358430: pin jupyter-lsp to 1.5.1 to match docker-python; newer versions freeze the server
  "jupyter-lsp==1.5.1",
  # b/333854354: pin jupyter-server to 2.12.5; later versions break LSP
  "jupyter-server==2.12.5",
  "python-lsp-server",
  "jupyterlab-lsp",
  "jedi",
  "notebook>=6.5.4",
  "nbconvert>=6.5.1",
  "traitlets==5.9.0",
  "jupyter-bokeh>=4.0.5",
]
docs = [
  "chess>=1.11.2",
  "emoji>=2.14.1",
  "jupyter-cache>=1.0.1",
  "pronouncing>=0.2.0",
  "roman>=5.0",
  "sentence-transformers>=4.1.0",
  "syllables>=1.0.9",
  "moviepy>=2.2.1",               # for doom example
  "vizdoom>=1.2.4",
]
dev = [
  { include-group = "test" },
  { include-group = "kaggle" },
  { include-group = "docs" },
  "mypy",
  "ruff",
  "types-tqdm>=4.67.0.20241221",
  "pandas-stubs>=2.2.3.241126",
  "docker-stubs>=0.1.0",
  "types-protobuf<6.0",
  "pre-commit>=4.2.0",
  "mypy-protobuf>=3.6.0",
]

[tool.uv]
# Supply chain hardening: refuse to install any package published after this date.
# To upgrade packages: bump this date to ~2 days before today, then run `uv lock --upgrade`.
# This is a fixed date (not a rolling window) and must be bumped periodically.
# See: https://docs.astral.sh/uv/reference/settings/#exclude-newer
exclude-newer = "2026-03-24T00:00:00Z"

[tool.mypy]
warn_unused_configs = true

disable_error_code = ["import-untyped"]
exclude = ["src/benchmarks/protos/utils"]

[tool.ruff]
extend-exclude = ["**/*_pb2.py", "src/kaggle_secrets.py", "src/kaggle_web_client.py"]

[tool.ruff.lint]
extend-select = ["I"]

[tool.ruff.lint.per-file-ignores]

"__init__.py" = ["F401", "F402"]