pyproject.toml

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

[tool.poetry]
name = "fast_dynamic_batcher"
version = "1.0.1"
description = "FastDynamicBatcher is a library for batching inputs across requests to accelerate machine learning workloads."
authors = ["Jeffrey Wigger"]
readme = "README.rst"
license = "MIT"
repository = "https://github.com/JeffWigger/FastDynamicBatcher"
homepage = "https://github.com/JeffWigger/FastDynamicBatcher"
keywords = ["machine-learning", "batching"]

[tool.poetry.group.dev.dependencies]
ruff = "~0.4.10"
mypy = "^1.10.0"
pytest = "^8.2.2"
pytest-asyncio = "0.23.7"
pytest-retry = "^1.6.3"
asgi-lifespan = "^2.1.0"
fastapi = "~0.115.3"
httpx = "^0.27.0"

[tool.poetry.group.test.dependencies]
torch = {url = "https://download.pytorch.org/whl/cpu/torch-2.3.1%2Bcpu-cp312-cp312-linux_x86_64.whl"}
torchvision = {url = "https://download.pytorch.org/whl/cpu/torchvision-0.18.1%2Bcpu-cp312-cp312-linux_x86_64.whl#sha256=e234971d465e8ce481a2718375d9f0797b4381c2d79bb1a546a5f28e26620629"}

[tool.poetry.dependencies]
python = ">=3.9"

[tool.mypy]
files = "setup.py"
python_version = "3.11"
strict = true
show_error_codes = true
enable_error_code = ["ignore-without-code", "redundant-expr", "truthy-bool"]
warn_unreachable = true

[[tool.mypy.overrides]]
module = ["ninja"]
ignore_missing_imports = true


[tool.pytest.ini_options]
minversion = "6.0"
addopts = ["-ra", "--showlocals", "--strict-markers", "--strict-config"]
xfail_strict = true
filterwarnings = ["error"]
testpaths = ["tests"]

[tool.cibuildwheel]
test-command = "pytest {project}/tests"
test-extras = ["test"]
test-skip = ["*universal2:arm64"]
# Setuptools bug causes collision between pypy and cpython artifacts
before-build = "rm -rf {project}/build"

[tool.ruff]
extend-select = [
  "B",    # flake8-bugbear
  "B904",
  "I",    # isort
  "PGH",  # pygrep-hooks
  "RUF",  # Ruff-specific
  "UP",   # pyupgrade
]
extend-ignore = [
  "E501",   # Line too long
]
target-version = "py39"
line-length = 120

[tool.ruff.lint]
select = ["E4", "E7", "E9", "F"]
ignore = []
fixable = ["ALL"]
unfixable = []
dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$"


[tool.ruff.format]
quote-style = "double"
indent-style = "space"
skip-magic-trailing-comma = false
line-ending = "auto"

[tool.ruff.lint.isort]
lines-after-imports = 2
lines-between-types = 1
known-first-party = ["fast_dynamic_batcher"]