File size: 732 Bytes
78000ed |
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 |
import nox
LOCATIONS = ("src/", "tests/", "noxfile.py", "setup.py")
@nox.session
def lint(session):
session.install("flake8")
session.install("flake8-bugbear")
session.install("flake8-isort")
session.install("black==24.3.0")
args = session.posargs or LOCATIONS
session.run("flake8", *args)
session.run("black", "--check", "--diff", *args)
@nox.session(python=["3.8", "3.9", "3.10", "3.11", "3.12"])
def tests(session):
session.install(
"torch==2.2.1",
"torchvision",
"--index-url",
"https://download.pytorch.org/whl/cpu",
)
session.install(".")
session.install("pytest")
session.install("pytest-mock")
session.run("pytest", *session.posargs)
|