Merge branch 'improve-ptemcee' of git.ligo.org:Monash/tupak into improve-ptemcee

bilby/core/sampler/ptemcee.py

@@ -89,6 +89,7 @@ class Ptemcee(MCMCSampler):
 
     def get_pos0_from_prior(self):
         """ for ptemcee, the pos0 has the shape ntemps, nwalkers, ndim """
+        logger.info("Generating pos0 samples")
         return [[self.get_random_draw_from_prior()
                  for _ in range(self.sampler_init_kwargs["nwalkers"])]
                 for _ in range(self.kwargs['ntemps'])]
@@ -120,6 +121,7 @@ class Ptemcee(MCMCSampler):
     def run_sampler(self):
         import schwimmbad
         if self.threads > 1:
+            logger.info("Creating MultiPool with {} processes".format(self.threads))
             with schwimmbad.MultiPool(self.threads) as pool:
                 self.pool = pool
                 return self.run_sampler_internal()
@@ -135,6 +137,7 @@ class Ptemcee(MCMCSampler):
         self.tau_list_n = []
 
         t0 = datetime.datetime.now()
+        logger.info("Starting to sample")
         for (pos0, lnprob, lnlike) in sampler.sample(
                 pos0, **self.sampler_function_kwargs):
             # Only check convergence every ncheck steps