49 lines
982 B
TOML
49 lines
982 B
TOML
[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"]
|