Add orbax/checkpoint/_src/BUILD

PiperOrigin-RevId: 714844457
google · Jan 13, 2025 · cab1e10 · cab1e10
1 parent ea00fa3
commit cab1e10
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/checkpoint/orbax/checkpoint/_src/BUILD b/checkpoint/orbax/checkpoint/_src/BUILD
@@ -0,0 +1,26 @@
+package(
+    default_applicable_licenses = ["//:package_license"],
+    default_visibility = ["//visibility:public"],
+)
+
+py_library(
+    name = "asyncio_utils",
+    srcs = ["asyncio_utils.py"],
+)
+
+py_test(
+    name = "asyncio_utils_test",
+    srcs = ["asyncio_utils_test.py"],
+    deps = [":asyncio_utils"],
+)
+
+py_library(
+    name = "composite",
+    srcs = ["composite.py"],
+)
+
+py_test(
+    name = "composite_test",
+    srcs = ["composite_test.py"],
+    deps = [":composite"],
+)
diff --git a/requirements.txt b/requirements.txt
@@ -20,4 +20,5 @@ etils[epath]
 simplejson
 chex
 optax
+nest_asyncio
-Original file line number
+Diff line change
@@ Expand Up / @@ -20,4 +20,5 @@ etils[epath] @@
     simplejson
     chex
     optax
+    nest_asyncio