Add /checkpoint/_src/path/snapshot/BUILD and /checkpoint/_src/path/BU…

…ILD.

PiperOrigin-RevId: 699277104

BUILD

@@ -1,3 +1,5 @@
 package(default_visibility = ["//visibility:private"])
+
 licenses(["notice"])
-exports_files(["LICENSE"])
+
+exports_files(["LICENSE"])

checkpoint/orbax/checkpoint/_src/path/BUILD

@@ -0,0 +1,21 @@
+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__",
+    ],
+)
+
+py_library(
+    name = "utils",
+    srcs = ["utils.py"],
+    srcs_version = "PY3",
+)
+
+py_library(
+    name = "path",
+    srcs = ["__init__.py"],
+    lib_rule = pytype_strict_library,
+)

checkpoint/orbax/checkpoint/_src/path/snapshot/BUILD

@@ -0,0 +1,30 @@
+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__",
+    ],
+)
+
+py_library(
+    name = "snapshot",
+    srcs = ["snapshot.py"],
+    srcs_version = "PY3",
+    deps = [
+        "//pyglib:timer",
+        "//third_party/py/orbax/checkpoint/_src/path:utils",
+    ],
+)
+
+py_test(
+    name = "snapshot_test",
+    srcs = ["snapshot_test.py"],
+    srcs_version = "PY3",
+    deps = [
+        ":snapshot",
+        "//absl/testing:absltest",
+    ],
+)