pyproject.toml

[project]
name = "datashare-python"
version = "0.1.3"
description = "Implement Datashare task in Python"
authors = [
    { name = "Clément Doumouro", email = "cdoumouro@icij.org" },
    { name = "Clément Doumouro", email = "clement.doumouro@gmail.com" }
]
readme = "README.md"
requires-python = "~=3.11"
dependencies = [
    "aiostream~=0.6.4",
    "aiohttp~=3.11.9",
    "icij-common[elasticsearch]~=0.5.5",
    "icij-worker[amqp]~=0.13",
    "torch==2.6.0.dev20241101 ; sys_platform != 'darwin'",
    "torch!=2.6.0.dev20241101+cpu,<=2.6.0.dev20241101; sys_platform == 'darwin'", # see https://github.com/astral-sh/uv/issues/9209
    "transformers~=4.46.3",
    "pycountry>=24.6.1",
    "sentencepiece>=0.2.0",
    "typer>=0.13.1",
    "alive-progress>=3.2.0",
]

[project.urls]
Homepage = "https://icij.github.io/datashare-python/"
Documentation = "https://icij.github.io/datashare-python/"
Repository = "https://github.com/ICIJ/datashare-python"
Issues = "https://github.com/ICIJ/datashare-python/issues"

[project.scripts]
datashare-python = "datashare_python.__main__:main"

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

[tool.uv]
package = true

[tool.uv.sources]
torch = [
    # TODO: add CUDA support
    { index = "pytorch-nightly" },
]

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

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

[dependency-groups]
dev = [
    "pytest~=8.1",
    "pytest-asyncio~=0.24",
    "pylint~=3.1.0",
    "psutil>=6.1.0",
    "redis[hiredis]>=5.2.1",
    "nest-asyncio>=1.6.0",
]
doc = [
    "mkdocs-material~=9.5.48",
    "termynal~=0.12.2",
]

[tool.uv.pip]
prerelease = "if-necessary"

[tool.pytest.ini_options]
asyncio_mode = "auto"
asyncio_default_fixture_loop_scope = "session"
markers = [
    "integration",
    "pull",
]