ENH: add warning about max steps

mj-will · Oct 23, 2024 · 7929973 · 7929973
1 parent f4883fc
commit 7929973
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/nessai/experimental/proposal/mcmc/proposal.py b/nessai/experimental/proposal/mcmc/proposal.py
@@ -144,6 +144,14 @@ def populate(
             z_chain[i] = z_current
             if self.n_accept is not None and n_accept.mean() > self.n_accept:
                 break
+        else:
+            if self.n_accept is not None:
+                logger.warning(
+                    (
+                        f"Reached max steps ({self.n_steps}) with "
+                        f"n_accept={self.n_accept}!"
+                    )
+                )
 
         z_new_history = np.array(z_new_history)
         self.step.update_stats(