No public description

PiperOrigin-RevId: 616621754
tensorflow · Mar 17, 2024 · 8bcf395 · 8bcf395
1 parent 90331a1
commit 8bcf395
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 16 deletions.
diff --git 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
@@ -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",
     ],