mirror of
https://github.com/microsoft/autogen.git
synced 2026-04-20 03:02:16 -04:00
fix delete (#950)
This commit is contained in:
committed by
GitHub
parent
27b2712016
commit
a3e770eac5
@@ -732,10 +732,8 @@ class TransformersEstimator(BaseEstimator):
|
||||
|
||||
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()
|
||||
|
||||
Reference in New Issue
Block a user