pyproject.toml

[build-system]
requires = ["setuptools>=77", "setuptools_scm>=8"]

build-backend = "setuptools.build_meta"

[project]
name = "quantizers"
authors = [{ name = "Chang Sun", email = "chsun@cern.ch" }]
description = ""
readme = "README.md"
requires-python = ">=3.10"
license = "LGPL-3.0-only"
license-files = ["LICENSE"]
keywords = ["quantization"]

classifiers = [
  "Programming Language :: Python :: 3",
  "Operating System :: OS Independent",
  "Development Status :: 5 - Production/Stable",
]
dynamic = ["version"]


[project.urls]
repository = "https://github.com/calad0i/quantizers"


[tool.setuptools]
include-package-data = true

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


[tool.setuptools_scm]
write_to = "src/quantizers/_version.py"

[project.optional-dependencies]
keras = ["keras>=3.0.0"]
torch = ["torch>=2.12"]
test = [
  'pytest',
  'pytest-cov',
  'pytest-xdist',
  'pytest-env',
  'jax',
  'tensorflow',
  'torch',
  'tinygrad',
  'mlx[cpu]',
  'keras>=3.0.0',
  'triton; sys_platform == "linux"',
]

[tool.pytest.ini_options]
filterwarnings = []

env = ["JAX_PLATFORMS=cpu", "CUDA_VISIBLE_DEVICES=-1", "TRITON_INTERPRET=1"]

pythonpath = ["test"]
addopts = "--import-mode=importlib --cov=src/quantizers --cov-report=xml --cov-report=html"

[tool.coverage.run]
omit = ["*/_version.py"]
patch = ["subprocess"]

[tool.coverage.report]
exclude_also = [
  '^[ \t]*\.\.\.[ \t]*(?:#.*)?$',
  '^[ \t]*raise NotImplementedError\b',
]
[tool.ruff]
include = ["src/**/*.py"]

line-length = 130
indent-width = 4

target-version = "py310"

[tool.ruff.format]
quote-style = "single"
indent-style = "space"
skip-magic-trailing-comma = false
docstring-code-format = true
docstring-code-line-length = 130

[tool.ruff.lint]
select = ["E", "W", "F", "I", "F401"]
ignore = ["E501", "F403", "F405", "E741"]
explicit-preview-rules = true
fixable = ["ALL"]
unfixable = []