pyproject.toml

[project]
name = "vortexsplit"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.10,<3.14"
dependencies = [
    "huggingface-hub>=0.19.0",
    "pillow>=10.0.0",
    "torch>=2.7.0,<2.8",
    "torchvision>=0.22.0,<0.23",
    "transformers==4.35.2",
    "flmr",
    # ColBERT (vendored under extern/FLMR/third_party) declares no deps in its
    # setup.py, so its runtime requirements are listed here explicitly.
    "colbert",
    "ujson",
    "bitarray",
    "gitpython",
    "ninja",
    "scipy",
    "tqdm",
    "easydict",
    # datasets must be >=4.0: the M2KR dataset schema uses the `List` feature
    # type, which older datasets can't parse. (4.x returns a lazy `Column` for
    # ds[col]; we convert to list at the call site in the indexing script.)
    "datasets>=4.8.5",
    # Community GPU faiss wheel matching the cu118 torch stack (official
    # faiss-gpu is conda-only). Provides colbert's faiss backend.
    "faiss-gpu-cu11",
    # Renders FX / module dataflow graphs to SVG (needs system graphviz `dot`).
    "pydot",
    # NVIDIA Management Library bindings (official, provides `import pynvml`) —
    # device-memory sampling during profiling.
    "nvidia-ml-py",
    "matplotlib>=3.10.9",
]

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

[tool.uv.sources]
torch = { index = "pytorch-cu118" }
torchvision = { index = "pytorch-cu118" }
torchaudio = { index = "pytorch-cu118" }
flmr = { path = "extern/FLMR", editable = true }
colbert = { path = "extern/FLMR/third_party/ColBERT", editable = true }

[tool.pytest.ini_options]
# Only collect this project's tests; never recurse into the vendored
# extern/FLMR (and its third_party/ColBERT) test suites.
testpaths = ["tests"]
markers = [
    "slow: end-to-end tests that load the PreFLMR/CLIP checkpoints and run a forward",
]

[tool.ruff]
line-length = 120
target-version = "py310"
# Never lint/format the vendored FLMR + ColBERT sources (mirrors pytest's testpaths).
extend-exclude = ["extern", ".torchsplit"]

[tool.ruff.lint]
select = ["E", "F", "I", "UP"]

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

[tool.hatch.build.targets.wheel]
packages = ["vortexsplit"]

[dependency-groups]
dev = [
    "datasets>=4.8.5",
    "pytest>=9.0.3",
    "ruff>=0.15.15",
]