pyproject.toml

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

[project]
name = "fasterbench"
dynamic = ["version"]
description = "A library for benchmarking AI models"
readme = "README.md"
requires-python = ">=3.7"
license = {text = "Apache-2.0"}
authors = [{name = "nathanhubens", email = "nathan.hubens@gmail.com"}]
keywords = ['benchmark', 'deep-learning', 'pytorch', 'profiling', 'latency', 'energy']
classifiers = ["Natural Language :: English", "Intended Audience :: Developers", "Development Status :: 4 - Beta", "Programming Language :: Python :: 3", "Programming Language :: Python :: 3 :: Only"]
dependencies = ['torch', 'fastcore', 'prettytable', 'tqdm', 'codecarbon', 'torchprofile', 'thop', 'plotly']

[project.urls]
Repository = "https://github.com/FasterAI-Labs/fasterbench"
Documentation = "https://FasterAI-Labs.github.io/fasterbench"

[project.entry-points.nbdev]
fasterbench = "fasterbench._modidx:d"

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

[tool.setuptools.packages.find]
include = ["fasterbench"]

[tool.nbdev]
version = "0.1.0"
tst_flags = 'notest'
jupyter_hooks = true