gnn/tensorflow_gnn/runner/tasks/BUILD at 1b152b2fe56ea7387b6e9bdd4967b2c95c19b3c9 · tensorflow/gnn · 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
load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "pytype_strict_library")
load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "py_strict_test")
load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "distribute_py_test")

licenses(["notice"])

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

py_strict_test(
    name = "classification_test",
    srcs = ["classification_test.py"],
    srcs_version = "PY3",
    tags = [
        "tf_2_10_and_up",  # TODO(b/237419537) remove on tf 2.10 release
    ],
    deps = [
        ":classification",
        "//:expect_absl_installed",
        "//tensorflow_gnn",
        "//tensorflow_gnn/runner:orchestration",
    ],
)

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

distribute_py_test(
    name = "dgi_test",
    srcs = ["dgi_test.py"],
    srcs_version = "PY3",
    xla_enable_strict_auto_jit = False,
    deps = [
        ":dgi",
        "//:expect_absl_installed",
        "//:expect_tensorflow_installed",
        "//:expect_tensorflow_installed:tensorflow_no_contrib",
        "//tensorflow_gnn",
        "//tensorflow_gnn/runner:orchestration",
    ],
)

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

py_strict_test(
    name = "regression_test",
    srcs = ["regression_test.py"],
    srcs_version = "PY3",
    tags = [
        "tf_2_10_and_up",  # TODO(b/237419537) remove on tf 2.10 release
    ],
    deps = [
        ":regression",
        "//:expect_absl_installed",
        "//:expect_tensorflow_installed",
        "//tensorflow_gnn",
        "//tensorflow_gnn/runner:orchestration",
    ],
)