DeepQuant/pyproject.toml at devel · pulp-platform/DeepQuant · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
# Copyright 2025 ETH Zurich and University of Bologna.
# Licensed under the Apache License, Version 2.0, see LICENSE for details.
# SPDX-License-Identifier: Apache-2.0
#
# Federico Brancasi <fbrancasi@ethz.ch>

[build-system]
requires = ["setuptools>=69.0.0", "wheel", "setuptools_scm>=6.2"]
build-backend = "setuptools.build_meta"

[project]
name = "DeepQuant"
version = "0.1.0"
description = "Brevitas model export utility for quantized networks"
requires-python = ">=3.11"
authors = [
    {name = "Victor J.B. Jung", email = "jungvi@iis.ee.ethz.ch"},
    {name = "Federico Brancasi", email = "fbrancasi@ethz.ch"},
]
dependencies = [
    "torch==2.1.2",
    "torchvision>=0.16.2",
    "torchaudio>=2.1.2",
    "brevitas==0.11.0",
    "torchmetrics",
    "black",
    "isort",
    "pytest",
    "netron",
    "tabulate",
    "tqdm",
    "colorama",
    "onnx",
    "onnxoptimizer",
    "onnxruntime",
]

[tool.setuptools]
packages = ["DeepQuant"]

[tool.pytest.ini_options]
python_files = ["Tests/*.py"]
python_functions = ["deepQuantTest*"]
markers = [
    "SingleLayerTests: Tests for individual layers",
    "ModelTests: Tests for full models",
]
filterwarnings = [
    "ignore::DeprecationWarning",
    "ignore::UserWarning"
]

[tool.black]
line-length = 88
target-version = ["py311"]

[tool.isort]
profile = "black"
multi_line_output = 3