pyproject.toml

[tool.poetry]
name = "huggingface-gradio-template"
version = "0.1.0"
description = "A boilerplate template for an IBM Granite Huggingface Spaces Gradio Demo"
authors = ["Martín Santillán Cooper <[email protected]>"]
license = "Apache-2.0"
readme = "README.md"
package-mode = false

[tool.poetry.dependencies]
python = ">=3.10,<3.11"
gradio = "5.9.1"
torch = "2.5.1"
spaces = "^0.30.4"
gradio-modal = "^0.0.4"
python-dotenv = "^1.0.1"
ibm-watsonx-ai = "^1.2.1"
numpy = ">=1.0.0,<2.0.0"
transformers = "^4.47.1"
accelerate = "^1.2.1"
vllm = "^0.7.3"

[tool.poetry.group.dev.dependencies]
pre-commit = "^4.0.1"
git-lint = "^0.1.2"
ruff = "^0.9.2"
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",
    "RUF015",
    "SIM117",
    "ANN001",
    "ANN201",
    "D103",
    "E501",
]
fixable = ["ALL"]
unfixable = []
line-length = 120


[tool.black]
line-length = 120

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