pyproject.toml

[project]
name = "trl-sandbox"
version = "0.0.0"
requires-python = "==3.12.*"

dependencies = [
  # https://github.com/huggingface/trl/issues/2377#issuecomment-2503234915
  # "accelerate>=1.7.0",
  "accelerate==0.34.2",
  "datasets>=3.6.0",
  "deepspeed>=0.17.1",
  "huggingface[cli]>=0.0.1",
  "peft>=0.15.2",
  # https://github.com/huggingface/trl/issues/2377#issuecomment-2503234915
  # "torch>=2.7.1",
  "torch==2.4.*",
  # https://github.com/huggingface/trl/issues/2377#issuecomment-2503234915
  # "transformers>=4.52.4",
  "transformers==4.50.*",
  "trl==0.18.*",
]

[tool.ruff]
target-version = "py39"
line-length = 119

[tool.ruff.lint]
ignore = [
  "B028", # warning without explicit stacklevel
  "C408", # dict() calls (stylistic)
  "C901", # function complexity
  "E501",
]
extend-select = ["E", "F", "I", "W", "UP", "B", "T", "C"]

[tool.ruff.lint.per-file-ignores]
# Allow prints in auxiliary scripts
"examples/**.py" = ["T201"]
"scripts/**.py" = ["T201"]
# Ignore import violations in all `__init__.py` files.
"__init__.py" = ["F401"]

[tool.ruff.lint.isort]
lines-after-imports = 2
known-first-party = ["trl"]

[tool.pytest.ini_options]
markers = [
  "slow: marks tests as slow (deselect with '-m \"not slow\"')",
  "low-priority: marks tests as low priority (deselect with '-m \"not low-priority\"')",
]