From 3d994feb1ed7ca1572e5ec84706e2161a80571bf Mon Sep 17 00:00:00 2001 From: Kai Fricke Date: Sat, 8 Jul 2023 18:21:00 -0700 Subject: [PATCH] [tune] Hotfix failing checkpoint test (#37220) #37173 changed a test in a previous iteration that is failing after additional changes. This PR reverts the changes to the test to fix broken master. Signed-off-by: Kai Fricke --- python/ray/tune/tests/test_function_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/tune/tests/test_function_api.py b/python/ray/tune/tests/test_function_api.py index 8312429ff4224..87ffc9de01d1b 100644 --- a/python/ray/tune/tests/test_function_api.py +++ b/python/ray/tune/tests/test_function_api.py @@ -296,7 +296,7 @@ def train(config, checkpoint_dir=None): checkpoint_obj = new_trainable2.save_to_object() new_trainable2.train() result = new_trainable2.train() - assert sum("tmp" in path for path in os.listdir(self.logdir)) == 0 + assert sum("tmp" in path for path in os.listdir(self.logdir)) == 1 new_trainable2.stop() assert sum("tmp" in path for path in os.listdir(self.logdir)) == 0 assert result[TRAINING_ITERATION] == 4