pyproject.toml

[tool.poetry]
name = "granite-vision-document-intelligence"
version = "0.1.0"
description = "Document intelligence demo powered by IBM Granite Vision and Docling"
authors = ["Pengyuan Li <pengyuan@ibm.com>"]
license = "Apache-2.0"
readme = "README.md"
package-mode = false

[tool.poetry.dependencies]
python = ">=3.10,<3.11"
gradio = "5.9.1"
torch = "2.4.0"
spaces = "0.30.4"
transformers = "^4.47.1"
accelerate = "^1.2.1"
PyMuPDF = ">=1.24.0"
Pillow = ">=10.0.0"
python-dotenv = "^1.0.0"
pandas = ">=2.0.0"
numpy = ">=1.24.0"
requests = ">=2.31.0"
tqdm = ">=4.66.0"
docling = ">=1.18.0"
docling-core = ">=2.5.0"


[tool.poetry.group.dev.dependencies]
pre-commit = "^4.0.1"
git-lint = "^0.1.2"
ruff = "^0.8.3"
pytest = "^8.3.4"


[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"


[tool.ruff]
select = [
    "E",   # pycodestyle
    "F",   # pyflakes
    "UP",  # pyupgrade
    "D",   # pydocstyle
    "I",   # isort
    "B",   # bugbear
    "ANN", # annotations
    "N",   # pep8-naming
    "C4",  # Comprehensions
    "DTZ", # DatetimeZ
    "Q",   # Quotes
    "SIM", # Simplify
    "RUF", # Ruff
]
ignore = ["D203", "D213"]
fixable = ["ALL"]
unfixable = []
line-length = 120


[tool.black]
line-length = 120

[tool.ruff.lint.pydocstyle]
convention = "google"