From ce89b97e8543a025fce6437e63ab6a2911d0e6a9 Mon Sep 17 00:00:00 2001 From: Graph Learning Team Date: Mon, 4 May 2026 14:02:46 -0700 Subject: [PATCH] Internal change. PiperOrigin-RevId: 910229689 --- tensorflow_gnn/models/gcn/BUILD | 1 + tensorflow_gnn/models/graph_sage/BUILD | 11 +++++++---- tensorflow_gnn/models/hgt/BUILD | 1 + tensorflow_gnn/models/mt_albis/BUILD | 1 + tensorflow_gnn/models/multi_head_attention/BUILD | 1 + tensorflow_gnn/models/vanilla_mpnn/BUILD | 1 + 6 files changed, 12 insertions(+), 4 deletions(-) diff --git a/tensorflow_gnn/models/gcn/BUILD b/tensorflow_gnn/models/gcn/BUILD index 83be83f4..8825eebd 100644 --- a/tensorflow_gnn/models/gcn/BUILD +++ b/tensorflow_gnn/models/gcn/BUILD @@ -8,6 +8,7 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test") licenses(["notice"]) package( + default_applicable_licenses = ["//tensorflow_gnn:license"], # copybara:strip default_visibility = [":__subpackages__"], ) diff --git a/tensorflow_gnn/models/graph_sage/BUILD b/tensorflow_gnn/models/graph_sage/BUILD index cdb28780..8f17efb1 100644 --- a/tensorflow_gnn/models/graph_sage/BUILD +++ b/tensorflow_gnn/models/graph_sage/BUILD @@ -7,10 +7,13 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test") licenses(["notice"]) -package(default_visibility = [ - "//tensorflow_gnn:__pkg__", - "//tensorflow_gnn/graph:__subpackages__", -]) +package( + default_applicable_licenses = ["//tensorflow_gnn:license"], # copybara:strip + default_visibility = [ + "//tensorflow_gnn:__pkg__", + "//tensorflow_gnn/graph:__subpackages__", + ], +) package_group(name = "users") diff --git a/tensorflow_gnn/models/hgt/BUILD b/tensorflow_gnn/models/hgt/BUILD index 354803e1..f39ef666 100644 --- a/tensorflow_gnn/models/hgt/BUILD +++ b/tensorflow_gnn/models/hgt/BUILD @@ -11,6 +11,7 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test") licenses(["notice"]) package( + default_applicable_licenses = ["//tensorflow_gnn:license"], # copybara:strip default_visibility = [":__subpackages__"], ) diff --git a/tensorflow_gnn/models/mt_albis/BUILD b/tensorflow_gnn/models/mt_albis/BUILD index 4ee7c049..11c121ef 100644 --- a/tensorflow_gnn/models/mt_albis/BUILD +++ b/tensorflow_gnn/models/mt_albis/BUILD @@ -11,6 +11,7 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test") licenses(["notice"]) package( + default_applicable_licenses = ["//tensorflow_gnn:license"], # copybara:strip default_visibility = [":__subpackages__"], ) diff --git a/tensorflow_gnn/models/multi_head_attention/BUILD b/tensorflow_gnn/models/multi_head_attention/BUILD index e22db374..f0dd732a 100644 --- a/tensorflow_gnn/models/multi_head_attention/BUILD +++ b/tensorflow_gnn/models/multi_head_attention/BUILD @@ -11,6 +11,7 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test") licenses(["notice"]) package( + default_applicable_licenses = ["//tensorflow_gnn:license"], # copybara:strip default_visibility = [":__subpackages__"], ) diff --git a/tensorflow_gnn/models/vanilla_mpnn/BUILD b/tensorflow_gnn/models/vanilla_mpnn/BUILD index fdb8417b..1aaf7b85 100644 --- a/tensorflow_gnn/models/vanilla_mpnn/BUILD +++ b/tensorflow_gnn/models/vanilla_mpnn/BUILD @@ -11,6 +11,7 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test") licenses(["notice"]) package( + default_applicable_licenses = ["//tensorflow_gnn:license"], # copybara:strip default_visibility = [":__subpackages__"], )