pyproject.toml

[build-system]
requires = ["setuptools>=45", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "medrax"
version = "0.1.0"
description = "A medical imaging and natural language processing project"
authors = [
    {name = "Adibvafa Fallahpour", email = "adibvafa.fallahpour@vectorinstitute.ai"},
]
license = {file = "LICENSE"}
requires-python = ">=3.10"
dependencies = [
    "requests==2.31.0",
    "numpy==1.26.4",
    "langchain-core==0.3.35",
    "langchain-openai==0.3.5",
    "langgraph==0.2.71",
    "python-dotenv==1.0.0",
    "pandas==2.2.3",
    "pydantic==2.5.2",
    "Pillow==10.1.0",
    "torchxrayvision==1.3.2",
    "transformers==4.34.0",
    "tokenizers==0.15.3",
    "sentencepiece==0.1.99",
    "accelerate==0.8.0",
    "markdown2[all]>=2.4.8",
    "protobuf==4.25.3",
    "gradio==5.16.0",
    "gradio_client==1.7.0",
    "httpx==0.28.1",
    "uvicorn==0.34.0",
    "fastapi==0.115.8",
    "einops==0.8.1",
    "tiktoken==0.8.0",
    "openai==1.62.0",
    "backoff==2.2.1",
    "torch==2.6.0.dev20241112",
    "torchvision==0.20.0.dev20241113",
    "scikit-image==0.22.0",
    "opencv-python==4.11.0.86",
    "matplotlib==3.10.0",
    "diffusers==0.31.1",
    "pydicom==3.0.1",
    "albumentations==2.0.4",
    "pyarrow==13.0.0",
]

[project.optional-dependencies]
dev = [
    "pytest",
    "black",
    "isort",
    "flake8",
    "mypy",
]

[tool.setuptools.packages.find]
where = ["."]
include = ["medrax*"]
exclude = ["med_env*", "demo*"]

[tool.black]
line-length = 100
target-version = ['py310']

[tool.isort]
profile = "black"
line_length = 100

[tool.mypy]
ignore_missing_imports = true
strict_optional = true

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