pyproject.toml

[build-system]
requires = ["flit_core >=3.2,<4"]
build-backend = "flit_core.buildapi"

[project]
name = "nygaardcodecommentclassification"
version = "0.0.1"
description = "This task is about Code Comment Classification for 3 different programming languages. This task is inspired by a challenge and will be used for the SEAES course"
authors = [
  { name = "Your name (or your organization/company/team)" },
]

classifiers = [
    "Programming Language :: Python :: 3",
    
]
dependencies = []
requires-python = "~=3.12.0"

[project.optional-dependencies]
api = [
    "catboost>=1.2.8",
    "dagshub>=0.6.3",
    "fastapi>=0.115.0",
    "joblib>=1.4.0",
    "loguru",
    "mlflow==2.22.0",
    "numpy>=1.26.0",
    "python-multipart>=0.0.20",
    "setfit>=1.1.3",
    "torch>=2.7.0",
    "uvicorn>=0.32.0",
    "python-dotenv",
]

dev = [
    "mkdocs",
    "pre-commit>=4.3.0",
    "pytest",
    "ruff",
]

training = [
    "datasets==4.3.0",
    "deepchecks[nlp]>=0.19.1",
    "optuna>=4.5.0",
    "tqdm",
    "typer",
]

frontend = [
    "gradio>=5.0.0",
    "requests",
]

all = [
    "nygaardcodecommentclassification[api,dev,training,frontend]",
]

[tool.uv]
override-dependencies = ["pyarrow==21.0.0"]

[tool.ruff]
line-length = 99
src = ["nygaardcodecommentclassification"]
include = ["pyproject.toml", "nygaardcodecommentclassification/**/*.py", "tests/**/*.py"]

[tool.ruff.lint]
extend-select = ["I"]  # Add import sorting
# Enable all `pydocstyle` rules, limiting to those that adhere to the
# Google convention via `convention = "google"`, below.
select = ["D"]

[tool.ruff.lint.isort]
known-first-party = ["nygaardcodecommentclassification"]
force-sort-within-sections = true

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

[tool.uv.sources]
torch = [
    { index = "pytorch-gpu" },
]

[[tool.uv.index]]
name = "pytorch-gpu"
url = "https://download.pytorch.org/whl/cu130"
explicit = true