code/pyproject.toml

# Python package of the SuperPoint tt-nn port:  pip install -e code/   (or  pip install code/)
# Install it into an environment that already has ttnn / tt-metal (ttnn is not a PyPI dependency here).
[build-system]
requires = ["setuptools>=64"]
build-backend = "setuptools.build_meta"

[project]
name = "tt-superpoint"
version = "0.1.0"
description = "SuperPoint keypoint detector and descriptor on a Tenstorrent Blackhole chip (tt-nn)"
readme = "PYTHON.md"
requires-python = ">=3.10"
license = { text = "Apache-2.0 (code); weights: Magic Leap SuperPoint licence, noncommercial research only" }
dependencies = [
    "numpy",
    "torch",
    "pillow",
    "transformers>=4.40",
    "huggingface_hub",
    "safetensors",
]

[project.optional-dependencies]
server = ["fastapi", "uvicorn", "pydantic"]
test = ["pytest"]

[project.urls]
"Model card" = "https://huggingface.co/changh95/superpoint-p150"
"Port source" = "https://github.com/changh95/tt-superpoint"

[tool.setuptools]
# tt_superpoint        public API (code/tt_superpoint)
# tt_superpoint._port  the port package code/models (the HTTP server keeps importing it as `models`)
# tt_superpoint.kernels  the custom device kernel sources (code/kernels), compiled at run time
package-dir = { "tt_superpoint" = "tt_superpoint", "tt_superpoint._port" = "models", "tt_superpoint.kernels" = "kernels" }
packages = [
    "tt_superpoint",
    "tt_superpoint._port",
    "tt_superpoint._port.reference",
    "tt_superpoint._port.tt",
    "tt_superpoint.kernels",
]
include-package-data = false

[tool.setuptools.package-data]
"tt_superpoint.kernels" = ["**/*.cpp", "**/*.hpp", "**/*.h", "**/*.inc"]