Skip to content

Commit

Permalink
[train] New persistence mode cleanup: Remove legacy checkpoint manage…
Browse files Browse the repository at this point in the history
…ment internals (#40069)

This PR removes the unused legacy checkpoint managers (`air._internal._CheckpointManager`, `tune.execution.CheckpointManager`). This also involved removing `_TrackedCheckpoint` dependencies and removing the `CheckpointStorage.MEMORY` checkpoint type throughout the code.

---------

Signed-off-by: Justin Yu <justinvyu@anyscale.com>
  • Loading branch information
justinvyu committed Oct 5, 2023
1 parent f78b95e commit 3152779
Show file tree
Hide file tree
Showing 13 changed files with 108 additions and 1,631 deletions.
8 changes: 0 additions & 8 deletions python/ray/air/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -49,14 +49,6 @@ py_test(
deps = [":ml_lib"]
)

py_test(
name = "test_checkpoint_manager",
size = "small",
srcs = ["tests/test_checkpoint_manager.py"],
tags = ["team:ml", "exclusive"],
deps = [":ml_lib"]
)

py_test(
name = "test_configs",
size = "small",
Expand Down
Loading

0 comments on commit 3152779

Please sign in to comment.