flops-profiler/pyproject.toml at main · cli99/flops-profiler · 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
[tool.poetry]
name = "flops-profiler"
version = "0.1.2"
description = "Measures the parameters, latency, and floating-point operations of PyTorch model."
authors = ["Cheng Li <pistasable@gmail.com>"]
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.8"
importlib-metadata = "^6.0.0"

[tool.poetry.dev-dependencies]
pytest = "^7.2.1"
flake8-docstrings = "^1.7.0"
darglint = "^1.8.1"
Sphinx = "^6.1.3"
sphinx-autodoc-typehints = "^1.23.4"
coverage = {version = "^7.1.0"}
pytest-cov = "^4.0.0"

[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"

[tool.coverage.paths]
source = ["flops_profiler", "*/site-packages"]
[tool.coverage.run]
branch = true
source = ["flops_profiler"]
[tool.coverage.report]
show_missing = true