From ea330b8b060773994312b9a41d5e4bff8e6ea3c0 Mon Sep 17 00:00:00 2001
From: Yaning Liang <yaning@google.com>
Date: Fri, 22 Nov 2024 14:22:55 -0800
Subject: [PATCH] fix BUILD files.

PiperOrigin-RevId: 699287624
---
 BUILD                                           | 12 +++++++++---
 checkpoint/orbax/checkpoint/_src/path/BUILD     | 12 ++++--------
 .../orbax/checkpoint/_src/path/snapshot/BUILD   | 17 +++++------------
 3 files changed, 18 insertions(+), 23 deletions(-)

diff --git a/BUILD b/BUILD
index 1395e33e0..f42af0049 100644
--- a/BUILD
+++ b/BUILD
@@ -1,5 +1,11 @@
-package(default_visibility = ["//visibility:private"])
+package(default_visibility = ["//visibility:public"])
 
-licenses(["notice"])
+licenses(
+    ["notice"],
+    default_visibility = ["//visibility:public"],
+)
 
-exports_files(["LICENSE"])
+exports_files(
+    ["LICENSE"],
+    default_visibility = ["//visibility:public"],
+)
diff --git a/checkpoint/orbax/checkpoint/_src/path/BUILD b/checkpoint/orbax/checkpoint/_src/path/BUILD
index 5ac980230..2f3982364 100644
--- a/checkpoint/orbax/checkpoint/_src/path/BUILD
+++ b/checkpoint/orbax/checkpoint/_src/path/BUILD
@@ -1,21 +1,17 @@
-load("//devtools/python/blaze:pytype.bzl", "pytype_strict_library", pytype_strict_test = "pytype_strict_contrib_test")
-load("//devtools/python/blaze:strict.bzl", "py_library_providing_imports_info")
-
 package(
-    default_applicable_licenses = ["//third_party/py/orbax/checkpoint:package_license"],
-    default_visibility = [
-        "//third_party/py/orbax/checkpoint:__subpackages__",
-    ],
+    default_applicable_licenses = ["//:package_license"],
+    default_visibility = ["//visibility:public"],
 )
 
 py_library(
     name = "utils",
     srcs = ["utils.py"],
+    default_visibility = ["//visibility:public"],
     srcs_version = "PY3",
 )
 
 py_library(
     name = "path",
     srcs = ["__init__.py"],
-    lib_rule = pytype_strict_library,
+    default_visibility = ["//visibility:public"],
 )
diff --git a/checkpoint/orbax/checkpoint/_src/path/snapshot/BUILD b/checkpoint/orbax/checkpoint/_src/path/snapshot/BUILD
index 54cef2ce6..968dc98b0 100644
--- a/checkpoint/orbax/checkpoint/_src/path/snapshot/BUILD
+++ b/checkpoint/orbax/checkpoint/_src/path/snapshot/BUILD
@@ -1,27 +1,20 @@
-load("//devtools/python/blaze:pytype.bzl", "pytype_strict_library")
-load("//devtools/python/blaze:strict.bzl", "py_strict_test")
-
 package(
-    default_applicable_licenses = ["//third_party/py/orbax/checkpoint:package_license"],
-    default_visibility = [
-        "//learning/deepmind/jax/roc:__subpackages__",
-        "//third_party/py/orbax/checkpoint:__subpackages__",
-    ],
+    default_applicable_licenses = ["//:package_license"],
+    default_visibility = ["//visibility:public"],
 )
 
 py_library(
     name = "snapshot",
     srcs = ["snapshot.py"],
+    default_visibility = ["//visibility:public"],
     srcs_version = "PY3",
-    deps = [
-        "//pyglib:timer",
-        "//third_party/py/orbax/checkpoint/_src/path:utils",
-    ],
+    deps = ["//orbax/checkpoint/orbax/checkpoint/_src/path:utils"],
 )
 
 py_test(
     name = "snapshot_test",
     srcs = ["snapshot_test.py"],
+    default_visibility = ["//visibility:public"],
     srcs_version = "PY3",
     deps = [
         ":snapshot",