[project]
name = "zico"
version = "0.1.0"
description = "zico: a typed decision model - state + typed questions -> calibrated probability distributions"
requires-python = ">=3.10"
dependencies = [
    "torch",
    "transformers>=5.2",
    "datasets",
    "peft",
    "safetensors",
    "numpy",
    "pillow",  # images in the state
    "torchvision",  # transformers' image processor for the vision tower
]

[project.optional-dependencies]
dev = ["pytest", "httpx"]
serve = ["fastapi", "uvicorn"]
mlx = ["mlx", "mlx-lm"]  # Apple silicon serving backend

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

[tool.setuptools.packages.find]
include = ["zico*"]

[tool.setuptools.package-data]
zico = ["ui.html"]

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