[build-system]
requires = ["setuptools>=75"]
build-backend = "setuptools.build_meta"

[project]
name = "rara-worker"
version = "0.1.0"
description = "Python adaptation worker for RARA"
requires-python = ">=3.12"
dependencies = [
  "grpcio>=1.74,<2",
  "protobuf>=6.31,<7",
]

[project.optional-dependencies]
training = [
  "accelerate>=1.8,<2",
  "datasets>=4,<5",
  "peft>=0.17,<1",
  "transformers>=4.53,<5",
  "trl>=0.20,<1",
]
quantized = [
  "bitsandbytes>=0.46,<1; platform_system == 'Linux'",
]
dev = [
  "grpcio-tools>=1.74,<2",
  "pytest>=8.4,<9",
  "pytest-asyncio>=1.1,<2",
  "ruff>=0.12,<1",
]

[project.scripts]
rara-python-worker = "rara_worker.__main__:main"

[tool.setuptools.packages.find]
where = ["src"]

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

[tool.ruff]
line-length = 100
target-version = "py312"
extend-exclude = ["src/rara/v1/*_pb2.py", "src/rara/v1/*_pb2_grpc.py"]

[tool.ruff.lint]
select = ["E", "F", "I", "UP", "B", "SIM"]