pyproject.toml

[project]
name = "natsume-simple"
version = "0.3.0"
description = "simple dependency relation search engine for Japanese"
authors = [{ name = "Bor Hodošček", email = "dev@bor.space" }]
license = "MIT"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
    "datasets<3.2.0",
    "polars[pyarrow,excel]>=1.12.0",
    "ginza>=5.2.0",
    "ja-ginza>=5.2.0",
    # blocked on thinc (spaCy):
    "numpy>=1.24.3,<2.0.0",
    "spacy>=3.7.5",
    # "spacy-transformers>=1.2.5",
    # "torch>=2.5.1",
    "wtpsplit>=2.1.2",
    "duckdb>=1.1.3",
    "tqdm>=4.66.0",
    "pydantic>=2.10.5",
]

[tool.uv]
dev-dependencies = [
    "ipykernel>=6.29.5",
    "ipywidgets>=8.1.5",
    "jupyter>=1.1.1",
    "notebook>=7.3.2",
    "pytest-testmon>=2.1.1",
    "pytest-watch>=4.2.0",
    "pytest>=8.3.3",
    "ruff>=0.7.4",
]
conflicts = [
    [
        { extra = "cpu" },
        { extra = "cuda" },
        { extra = "rocm" },
    ],
]

[project.optional-dependencies]
cpu = [
    "torch>=2.5.1",
]
cuda = [
    "torch>=2.5.1",
    "onnxruntime-gpu>=1.20.1",
    "cupy-cuda12x>=13.3.0",
]
rocm = [
    "torch==2.5.1+rocm6.2",
    "pytorch_triton_rocm==3.1.0",
    "cupy-rocm-5-0",
]
backend = [
    "fastapi[standard]>=0.115.0",
    "python-fasthtml>=0.8.0",
]

[tool.uv.sources]
torch = [
    { index = "pytorch-cpu", extra = "cpu" },
    { index = "pytorch-cuda", extra = "cuda" },
    { index = "pytorch-rocm", extra = "rocm" },
    # { index = "pytorch-rocm", extra = "rocm", marker = "platform_system == 'Linux'" },
]
# pytorch-triton-rocm = { index = "pytorch-rocm", extra = "rocm", marker = "platform_system == 'Linux'" }
pytorch-triton-rocm = { index = "pytorch-rocm", extra = "rocm" }

[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true

[[tool.uv.index]]
name = "pytorch-rocm"
url = "https://download.pytorch.org/whl/rocm6.2"
explicit = true

[[tool.uv.index]]
name = "pytorch-cuda"
url = "https://download.pytorch.org/whl/cu124"
explicit = true

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
packages = ["src/natsume_simple"]

[tool.pytest.ini_options]
minversion = "6.0"
addopts = "-ra -q --doctest-modules"
testpaths = [
    "tests",
    "src",
]
doctest_optionflags = "NORMALIZE_WHITESPACE IGNORE_EXCEPTION_DETAIL"