From fbb4794db5f41f89f26b1f9f1a79d884f1058aff Mon Sep 17 00:00:00 2001 From: Felix Dangel Date: Mon, 9 Sep 2024 17:15:43 -0400 Subject: [PATCH] [DEL] Remove `get_resume_value` from API --- wandb_preempt/__init__.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/wandb_preempt/__init__.py b/wandb_preempt/__init__.py index c67e208..26ffec2 100644 --- a/wandb_preempt/__init__.py +++ b/wandb_preempt/__init__.py @@ -1,11 +1,8 @@ """wandb_preempt library.""" -from wandb_preempt.checkpointer import Checkpointer, get_resume_value +from wandb_preempt.checkpointer import Checkpointer -__all__ = [ - "Checkpointer", - "get_resume_value", -] +__all__ = ["Checkpointer"] # TODO Remove this function once we have a unit test that uses the checkpointer code