Skip to content

Commit

Permalink
Fix license issue.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 718436866
  • Loading branch information
liangyaning33 authored and Orbax Authors committed Jan 23, 2025
1 parent ad967e8 commit 3b5367f
Show file tree
Hide file tree
Showing 2 changed files with 42 additions and 4 deletions.
5 changes: 1 addition & 4 deletions checkpoint/orbax/checkpoint/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,7 @@ package(
default_visibility = ["//visibility:public"],
)

license(
name = "package_license",
package_name = "orbax-checkpoint",
)
licenses(["notice"])

py_library(
name = "checkpoint",
Expand Down
41 changes: 41 additions & 0 deletions checkpoint/orbax/checkpoint/_src/futures/BUILD
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
package(
default_applicable_licenses = ["//:package_license"],
default_visibility = ["//visibility:public"],
)

py_library(
name = "synchronization",
srcs = ["synchronization.py"],
deps = ["//checkpoint/orbax/checkpoint/_src/multihost"],
)

py_test(
name = "synchronization_test",
srcs = ["synchronization_test.py"],
deps = [
":synchronization",
"//checkpoint/orbax/checkpoint/_src/multihost",
],
)

py_library(
name = "future",
srcs = ["future.py"],
deps = [
":synchronization",
"//checkpoint/orbax/checkpoint/_src:asyncio_utils",
"//checkpoint/orbax/checkpoint/_src/multihost",
],
)

py_test(
name = "future_test",
srcs = ["future_test.py"],
args = ["--num_processes=1"],
python_version = "PY3",
deps = [
":future",
":synchronization",
"//checkpoint/orbax/checkpoint/_src/multihost",
],
)

0 comments on commit 3b5367f

Please sign in to comment.