PyTorch-FWD/noxfile.py at master · BonnBytes/PyTorch-FWD · 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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
"""Noxfile."""

import nox


@nox.session(name="format")
def format(session):
    """Fix common convention problems automatically."""
    session.install("black")
    session.install("isort")
    session.run("isort", "src", "noxfile.py")
    session.run("black", "src", "noxfile.py")
    session.run("isort", "scripts", "noxfile.py")
    session.run("black", "scripts", "noxfile.py")
    session.run("isort", "tests", "noxfile.py")
    session.run("black", "tests", "noxfile.py")


@nox.session(name="lint")
def lint(session):
    """Check code conventions."""
    session.install("flake8==4.0.1")
    session.install(
        "flake8-black",
        "flake8-docstrings",
        "flake8-bugbear",
        "flake8-broken-line",
        "pep8-naming",
        "pydocstyle",
        "darglint",
    )
    session.run("flake8", "src", "tests", "noxfile.py")


@nox.session(name="typing")
def mypy(session):
    """Check type hints."""
    session.install(".")
    session.install("mypy")
    session.run(
        "mypy",
        "--install-types",
        "--non-interactive",
        "--ignore-missing-imports",
        "--no-strict-optional",
        "--no-warn-return-any",
        "--implicit-reexport",
        "--allow-untyped-calls",
        "--explicit-package-bases",
        "src",
    )


@nox.session(name="test")
def test(session):
    """Run long pytest."""
    session.install(".")
    session.chdir("tests")
    # env handles deterministic CuBLAS with CUDA >= 10.2
    session.run("pytest", env={"CUBLAS_WORKSPACE_CONFIG": "4096:8"})


@nox.session(name="fast-test")
def run_test_fast(session):
    """Run pytest."""
    session.install(".")
    session.install("pytest")
    # env handles deterministic CuBLAS with CUDA >= 10.2
    session.run("pytest", "-m", "not slow", env={"CUBLAS_WORKSPACE_CONFIG": "4096:8"})


@nox.session(name="build")
def build(session):
    """Build a pip package."""
    session.install("build")
    session.run("python", "-m", "build")


@nox.session(name="finish")
def finish(session):
    """Finish this version increase the version number and upload to pypi."""
    session.install("bump2version")
    session.install("twine")
    session.run("bumpversion", "release", external=True)
    build(session)
    session.run("twine", "upload", "--skip-existing", "dist/*", external=True)
    session.run("git", "push", external=True)
    session.run("bumpversion", "patch", external=True)
    session.run("git", "push", external=True)


@nox.session(name="check-package")
def pyroma(session):
    """Run pyroma to check if the package is ok."""
    session.install("pyroma")
    session.run("pyroma", ".")