noxfile.py

"""
# Copyright (c) Facebook, Inc. and its affiliates.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
#
"""
# type: ignore
import nox


def install_torch_packages(session):
    session.install(
        "-f",
        "https://pytorch-geometric.com/whl/torch-1.4.0.html",
        "-f",
        "https://download.pytorch.org/whl/torch_stable.html",
        "-r",
        "requirements/torch.txt",
    )


@nox.session()
def lint(session):
    session.install("--upgrade", "setuptools", "pip")
    install_torch_packages(session)
    session.install("-r", "requirements/base.txt")
    session.install("-r", "requirements/dev.txt")
    session.run("flake8", "graphlog")
    session.run("black", "--check", "graphlog")
    session.run("mypy", "--strict", "graphlog")


@nox.session()
def test(session) -> None:
    session.install("--upgrade", "setuptools", "pip")
    install_torch_packages(session)
    session.install("-r", "requirements/base.txt")
    session.install("-r", "requirements/dev.txt")
    session.run("python", "-m", "pytest", "tests")