load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "pytype_strict_library")

licenses(["notice"])

pytype_strict_library(
    name = "runner",
    srcs = ["__init__.py"],
    visibility = ["//visibility:public"],
    deps = [
        ":orchestration",
        "//tensorflow_gnn/runner/input:datasets",
        "//tensorflow_gnn/runner/tasks:attribution",
        "//tensorflow_gnn/runner/tasks:classification",
        "//tensorflow_gnn/runner/tasks:dgi",
        "//tensorflow_gnn/runner/tasks:regression",
        "//tensorflow_gnn/runner/trainers:keras_fit",
        "//tensorflow_gnn/runner/utils:model_dir",
        "//tensorflow_gnn/runner/utils:strategies",
    ],
)

pytype_strict_library(
    name = "orchestration",
    srcs = ["orchestration.py"],
    srcs_version = "PY3",
    visibility = [":__subpackages__"],
    deps = [
        "//:expect_tensorflow_installed",
        "//tensorflow_gnn",
    ],
)
