From c847c753b9ae232bf7e1faefcec02274ecad10a7 Mon Sep 17 00:00:00 2001
From: Colm Talbot <colm.talbot@ligo.org>
Date: Wed, 18 Mar 2020 16:34:10 -0400
Subject: [PATCH] Improve log messages for dynesty resume
---
bilby/core/sampler/dynesty.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/bilby/core/sampler/dynesty.py b/bilby/core/sampler/dynesty.py
index 4c93f8efe..944f2f452 100644
--- a/bilby/core/sampler/dynesty.py
+++ b/bilby/core/sampler/dynesty.py
@@ -320,9 +320,9 @@ class Dynesty(NestedSampler):
def _run_external_sampler_with_checkpointing(self):
logger.debug("Running sampler with checkpointing")
if self.resume:
- resume = self.read_saved_state(continuing=True)
- if resume:
- logger.info('Resuming from previous run.')
+ resume_file_loaded = self.read_saved_state(continuing=True)
+ if resume_file_loaded:
+ logger.info('Resume file successfully loaded.')
old_ncall = self.sampler.ncall
sampler_kwargs = self.sampler_function_kwargs.copy()
@@ -366,8 +366,8 @@ class Dynesty(NestedSampler):
Whether the run is continuing or terminating, if True, the loaded
state is mostly written back to disk.
"""
- logger.info("Reading resume file {}".format(self.resume_file))
if os.path.isfile(self.resume_file):
+ logger.info("Reading resume file {}".format(self.resume_file))
with open(self.resume_file, 'rb') as file:
self.sampler = dill.load(file)
if self.sampler.added_live and continuing:
@@ -376,6 +376,7 @@ class Dynesty(NestedSampler):
self.sampler.rstate = np.random
self.start_time = self.sampler.kwargs.pop("start_time")
self.sampling_time = self.sampler.kwargs.pop("sampling_time")
+ return True
else:
logger.debug(
"Resume file {} does not exist.".format(self.resume_file))
--
GitLab