Skip to content

Commit d1df0e7

Browse files
committed
put checkpoint deletes in output_search_internal
1 parent 926a3f4 commit d1df0e7

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

autofit/non_linear/search/nest/dynesty/search/abstract.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -480,6 +480,12 @@ def search_internal_from(
480480
):
481481
raise NotImplementedError()
482482

483+
def output_search_internal(self, search_internal):
484+
try:
485+
os.remove(self.checkpoint_file)
486+
except TypeError:
487+
pass
488+
483489
def check_pool(self, uses_pool: bool, pool):
484490
if (uses_pool and pool is None) or (not uses_pool and pool is not None):
485491
raise exc.SearchException(

autofit/non_linear/search/nest/nautilus/search.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -464,6 +464,10 @@ def output_search_internal(self, search_internal):
464464
search_internal.pool_l = pool_l
465465
search_internal.pool_s = pool_s
466466

467+
if self.checkpoint_file is not None:
468+
469+
os.remove(self.checkpoint_file)
470+
467471
def samples_info_from(self, search_internal=None):
468472
return {
469473
"log_evidence": search_internal.evidence(),

0 commit comments

Comments
 (0)