[build-system]
requires = ["setuptools>=70", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "higher-order-pruner"
version = "0.1.0"
description = "Model-agnostic higher-order pruning for PyTorch and Hugging Face models"
readme = "README.md"
requires-python = ">=3.10"
license = {text = "Apache-2.0"}
authors = [{name = "Angshul Majumdar"}]
dependencies = [
  "torch>=2.4,<3",
  "transformers>=4.57,<5",
  "accelerate>=1.2,<2",
  "safetensors>=0.5,<1",
  "huggingface_hub>=0.34,<2",
]

[project.optional-dependencies]
test = ["pytest>=8,<10"]

[tool.setuptools.packages.find]
include = ["higher_order_pruner*"]

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