From 8bcf395b4967563b510d903cbc4ac6c2f54b56f7 Mon Sep 17 00:00:00 2001 From: Arno Eigenwillig Date: Sun, 17 Mar 2024 10:58:56 -0700 Subject: [PATCH] No public description PiperOrigin-RevId: 616621754 --- tensorflow_gnn/models/gcn/BUILD | 7 +------ tensorflow_gnn/models/mt_albis/BUILD | 13 +++---------- 2 files changed, 4 insertions(+), 16 deletions(-) diff --git a/tensorflow_gnn/models/gcn/BUILD b/tensorflow_gnn/models/gcn/BUILD index 40879933..ecdb586c 100644 --- a/tensorflow_gnn/models/gcn/BUILD +++ b/tensorflow_gnn/models/gcn/BUILD @@ -7,12 +7,7 @@ package( default_visibility = [":__subpackages__"], ) -package_group( - name = "users", - packages = [ - "//third_party/google_research/google_research/ugsl/...", - ], -) +package_group(name = "users") pytype_strict_library( name = "gcn", diff --git a/tensorflow_gnn/models/mt_albis/BUILD b/tensorflow_gnn/models/mt_albis/BUILD index 5accc114..98aa4446 100644 --- a/tensorflow_gnn/models/mt_albis/BUILD +++ b/tensorflow_gnn/models/mt_albis/BUILD @@ -7,16 +7,12 @@ package( default_visibility = [":__subpackages__"], ) -package_group(name = "users") - +# This is the public BUILD target for users of this model. pytype_strict_library( name = "mt_albis", srcs = ["__init__.py"], srcs_version = "PY3", - visibility = [ - ":__subpackages__", - ":users", - ], + visibility = ["//visibility:public"], deps = [ ":config_dict", ":layers", @@ -29,10 +25,7 @@ pytype_strict_library( pytype_strict_library( name = "hparams_vizier", srcs = ["hparams_vizier.py"], - visibility = [ - ":__subpackages__", - ":users", - ], + visibility = ["//visibility:public"], deps = [ "//:expect_vizier_service_pyvizier_installed", ],