Fix arm64 build failures

PiperOrigin-RevId: 729583824
google · Feb 21, 2025 · 2333a89 · 2333a89
1 parent 7aa5489
commit 2333a89
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/grain/_src/python/experimental/example_packing/BUILD b/grain/_src/python/experimental/example_packing/BUILD
@@ -9,9 +9,10 @@ py_library(
     srcs = ["packing.py"],
     srcs_version = "PY3",
     deps = [
+        "@abseil-py//absl/logging",
         "//grain/_src/core:tree_lib",
         "//grain/_src/python:record",
-        "@abseil-py//absl/logging",
+        "@pypi//jax:pkg",
         "@pypi//jaxtyping:pkg",
         "@pypi//numpy:pkg",
     ],