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

[project]
name = "hamer-p150"
version = "0.1.0"
description = "HaMeR hand mesh recovery (ViT-H/16 + MANO head) on a Tenstorrent Blackhole p150 via tt-nn"
readme = "README.md"
requires-python = ">=3.10"
license = { text = "MIT code; MANO model non-commercial (see the model card)" }
# ttnn / tt-metal is NOT declared: install this package into an environment that already has a built tt-metal
# (its python_env or a venv with ttnn).  Only the model's own host dependencies are listed.
dependencies = [
    "torch>=2.0",
    "numpy>=1.24.4,<2",
    "pillow>=9",
    "einops>=0.6,<1",
    "scipy>=1.10,<2",
    "huggingface_hub>=0.30,<2",
]

[project.optional-dependencies]
server = ["fastapi", "uvicorn", "pydantic>=2", "uvloop>=0.19,<1", "httptools>=0.6,<1"]
viz = ["matplotlib"]
test = ["pytest", "pyyaml"]  # host tests: also install [server] for the server-equality tests

[project.urls]
Model = "https://huggingface.co/changh95/hamer-p150"
Upstream = "https://github.com/geopavlakos/hamer"

[tool.setuptools.packages.find]
where = ["."]
include = ["hamer", "hamer.*"]
exclude = ["hamer.build*", "hamer.kernels*"]

[tool.setuptools.package-data]
hamer = ["kernels/*.cpp", "kernels/*/*.cpp", "kernels/*/*.hpp", "kernels/*/*.h", "native/*.c"]