pyproject.toml

[build-system]
requires = ["setuptools", "setuptools-scm"]
build-backend = "setuptools.build_meta"

[project]
name = "sleap-nn"
authors = [
    {name = "Divya Seshadri Murali", email = "dimurali@salk.edu"},
    {name = "Vincent Tu", email = "vitu@ucsd.edu"},
    {name = "Liezl Maree", email = "lmaree@salk.edu"},
    {name = "David Samy", email = "davidasamy@gmail.com"},
    {name = "Talmo Pereira", email = "talmo@salk.edu"}
]
description = "Neural network backend for training and inference for animal pose estimation."
requires-python = ">=3.9"
keywords = ["sleap", "pose estimation", "deep learning", "neural networks", "computer vision", "animal behavior"]
license = {text = "BSD-3-Clause"}
classifiers = [
    "Programming Language :: Python :: 3.9"
]
dependencies = [
    "torch",
    "torchvision",
    "pydantic",
    "lightning",
    "imageio",
    "imageio-ffmpeg",
    "av",
    "kornia",
    "hydra-core",
    "sleap-io>=0.1.0",
]
dynamic = ["version", "readme"]

[tool.setuptools.dynamic]
version = {attr = "sleap_nn.__version__"}
readme = {file = ["README.md"]}

[project.optional-dependencies]
dev = [
    "pytest",
    "pytest-cov",
    "black",
    "pydocstyle",
    "toml",
    "twine",
    "build",
    "ipython",
    "ruff"
]

[project.scripts]
biogtr = "sleap_nn.cli:cli"

[project.urls]
Homepage = "https://sleap.ai"
Repository = "https://github.com/talmolab/sleap-nn"

[tool.black]
line-length = 88

[tool.ruff.lint]
select = [
    "D",  # pydocstyle
]

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