pyproject.toml

[build-system]
requires = ["setuptools>=68"]
build-backend = "setuptools.build_meta"

[project]
name = "one-parameter-classifier"
version = "0.0.0"
description = "Image-level person classification on EUPE-ViT-B features with a single free parameter"
requires-python = ">=3.9"
dependencies = [
    "torch>=2.0",
    "numpy",
    "pillow",
    "safetensors",
    "transformers>=4.40",
    "huggingface-hub",
    "pycocotools",
]

[project.optional-dependencies]
dev = ["pytest>=7"]
synth = ["nosis"]

[tool.setuptools]
packages = ["common"]
py-modules = ["infer", "head", "verify", "calibrate", "rtl_gen", "synth"]

[tool.pytest.ini_options]
testpaths = ["tests"]