fix delete (microsoft#950)

DarylRodrigo · Mar 14, 2023 · a3e770e · a3e770e
1 parent 27b2712
commit a3e770e
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/flaml/automl/model.py b/flaml/automl/model.py
@@ -732,10 +732,8 @@ def on_epoch_end(self, args, state, control, **callback_kwargs):
 
     def _delete_one_ckpt(self, ckpt_location):
         if self._use_ray is False:
-            try:
+            if os.path.exists(ckpt_location):
                 shutil.rmtree(ckpt_location)
-            except FileNotFoundError:
-                logger.warning("checkpoint {} not found".format(ckpt_location))
 
     def cleanup(self):
         super().cleanup()