yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-11-02 02:40:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix delete (#950)

Browse Source
This commit is contained in:
Susan Xueqing Liu 2023-03-13 23:19:58 -04:00 committed by GitHub
parent 27b2712016
commit a3e770eac5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
flaml/automl/model.py
View File

@ -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()