remove checkpoint folder which is no longer being used.

PiperOrigin-RevId: 725643385
jax-ml · Feb 11, 2025 · a21537c · a21537c
1 parent 930dad6
commit a21537c
Show file tree

Hide file tree

Showing 5 changed files with 0 additions and 242 deletions.
diff --git a/jax_tpu_embedding/sparsecore/lib/BUILD b/jax_tpu_embedding/sparsecore/lib/BUILD
@@ -23,7 +23,6 @@ pytype_strict_library(
     srcs = ["__init__.py"],
     visibility = ["//jax_tpu_embedding/sparsecore:__pkg__"],
     deps = [
-        "//jax_tpu_embedding/sparsecore/lib/checkpointing",  # buildcleaner: keep
         "//jax_tpu_embedding/sparsecore/lib/core",  # buildcleaner: keep
         "//jax_tpu_embedding/sparsecore/lib/fdo",  # buildcleaner: keep
         "//jax_tpu_embedding/sparsecore/lib/flax",  # buildcleaner: keep

diff --git a/jax_tpu_embedding/sparsecore/lib/checkpointing/BUILD b/jax_tpu_embedding/sparsecore/lib/checkpointing/BUILD
diff --git a/jax_tpu_embedding/sparsecore/lib/checkpointing/__init__.py b/jax_tpu_embedding/sparsecore/lib/checkpointing/__init__.py
diff --git a/jax_tpu_embedding/sparsecore/lib/checkpointing/checkpoint_utils.py b/jax_tpu_embedding/sparsecore/lib/checkpointing/checkpoint_utils.py
diff --git a/jax_tpu_embedding/sparsecore/lib/checkpointing/checkpoint_utils_test.py b/jax_tpu_embedding/sparsecore/lib/checkpointing/checkpoint_utils_test.py