pyproject.toml

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

[project]
name = "TorchFix"
requires-python = ">=3.9"
description = "TorchFix - a linter for PyTorch-using code with autofix support"
readme = "README.md"
license = {file = "LICENSE"}
classifiers = [
  "Development Status :: 4 - Beta",
  "Programming Language :: Python"
]
dynamic = ["version"]
dependencies = [
  "flake8>=3.8.2",
  "PyYAML",
  "libcst>=1.5.0,<1.6.0"
]

[project.optional-dependencies]
dev = [
  "flake8==6.0.0",
  "pytest==7.2.0",
  "libcst==1.5.0",
  "types-PyYAML==6.0.7",
  "mypy==1.7.0",
  "black==24.4.0",
  "pre-commit==3.7.0",
]

[project.urls]
Repository = "https://github.com/pytorch-labs/torchfix"
"Bug Tracker" = "https://github.com/pytorch-labs/torchfix/issues"

[project.scripts]
torchfix = "torchfix.__main__:main"

[project.entry-points]
"flake8.extension" = {TOR = "torchfix.torchfix:TorchChecker"}

[tool.pytest.ini_options]
log_cli = true
log_cli_level = "INFO"
pythonpath = [
  "."
]

[tool.black]
exclude = "tests/fixtures/*"

[tool.setuptools.package-data]
"*" = ["*.yaml"]

[tool.mypy]
exclude = ["tests/fixtures", "build"]
check_untyped_defs = true

[tool.setuptools.dynamic]
version = {attr = "torchfix.torchfix.__version__"}