[tool.poetry]
name = "granite-vision-demos"
version = "0.1.0"
description = "A collection of gradio demos for granite vision models."
authors = ["Hendrik Strobelt <hendrik.strobelt@ibm.com>"]
license = "Apache-2.0"
readme = "README.md"
package-mode = false

[tool.poetry.dependencies]
python = ">=3.10,<3.11"
gradio = "5.16.1"
torch = "2.4.0"
spaces = "0.30.4"
transformers = { git = "https://github.com/huggingface/transformers" }
accelerate = "^1.2.1"
langchain-milvus = "^0.1.8"
langchain-core = "^0.3.36"
langchain-huggingface = "^0.1.2"
gradio-pdf = "^0.0.22"


[tool.poetry.group.dev.dependencies]
pre-commit = "^4.0.1"
git-lint = "^0.1.2"
#ruff = "^0.8.3"
ruff = "^0.9.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"