Internal

PiperOrigin-RevId: 705537829
google · Dec 13, 2024 · 20fc320 · 20fc320
1 parent 196d498
commit 20fc320
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/WORKSPACE b/WORKSPACE
@@ -1,6 +1,7 @@
 workspace(name = "org_orbax")
 
 load("@bazel_tools//tools/build_defs/repo:git.bzl", "git_repository")
+load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
 git_repository(
     name = "bazel_skylib",
@@ -13,3 +14,10 @@ git_repository(
     remote = "https://github.com/bazelbuild/rules_python.git",
     tag = "0.5.0",
 )
+
+http_archive(
+    name = "com_google_absl",
+    sha256 = "0320586856674d16b0b7a4d4afb22151bdc798490bb7f295eddd8f6a62b46fea",
+    strip_prefix = "abseil-cpp-fb3621f4f897824c0dbe0615fa94543df6192f30",
+    url = "https://github.com/abseil/abseil-cpp/archive/fb3621f4f897824c0dbe0615fa94543df6192f30.tar.gz",
+)
diff --git a/checkpoint/orbax/checkpoint/_src/path/snapshot/BUILD b/checkpoint/orbax/checkpoint/_src/path/snapshot/BUILD
@@ -16,6 +16,6 @@ py_test(
     srcs_version = "PY3",
     deps = [
         ":snapshot",
-        "//absl/testing:absltest",
+        "@com_google_absl//absl/testing:absltest",
     ],
 )
diff --git a/requirements.txt b/requirements.txt
@@ -13,4 +13,5 @@
 jax>=0.4.34
 jaxlib>=0.4.34
 portpicker~=1.6
+absl-py>=1.0,==1.*