Skip to content
Snippets Groups Projects
Commit e9e58ef0 authored by Colm Talbot's avatar Colm Talbot
Browse files

BUGFIX: Resolve "Matplotlib error at dynesty checkpoint plot"

parent c9bee354
No related branches found
No related tags found
1 merge request!1190BUGFIX: Resolve "Matplotlib error at dynesty checkpoint plot"
...@@ -669,10 +669,14 @@ class Dynesty(NestedSampler): ...@@ -669,10 +669,14 @@ class Dynesty(NestedSampler):
np.linalg.linalg.LinAlgError, np.linalg.linalg.LinAlgError,
ValueError, ValueError,
OverflowError, OverflowError,
Exception,
) as e: ) as e:
logger.warning(e) logger.warning(e)
logger.warning("Failed to create dynesty state plot at checkpoint") logger.warning("Failed to create dynesty state plot at checkpoint")
except Exception as e:
logger.warning(
f"Unexpected error {e} in dynesty plotting. "
"Please report at git.ligo.org/lscsoft/bilby/-/issues"
)
finally: finally:
plt.close("all") plt.close("all")
try: try:
...@@ -691,34 +695,53 @@ class Dynesty(NestedSampler): ...@@ -691,34 +695,53 @@ class Dynesty(NestedSampler):
np.linalg.linalg.LinAlgError, np.linalg.linalg.LinAlgError,
ValueError, ValueError,
OverflowError, OverflowError,
Exception,
) as e: ) as e:
logger.warning(e) logger.warning(e)
logger.warning("Failed to create dynesty unit state plot at checkpoint") logger.warning("Failed to create dynesty unit state plot at checkpoint")
except Exception as e:
logger.warning(
f"Unexpected error {e} in dynesty plotting. "
"Please report at git.ligo.org/lscsoft/bilby/-/issues"
)
finally: finally:
plt.close("all") plt.close("all")
try: try:
filename = f"{self.outdir}/{self.label}_checkpoint_run.png" filename = f"{self.outdir}/{self.label}_checkpoint_run.png"
fig, axs = dyplot.runplot( fig, _ = dyplot.runplot(
self.sampler.results, logplot=False, use_math_text=False self.sampler.results, logplot=False, use_math_text=False
) )
fig.tight_layout() fig.tight_layout()
plt.savefig(filename) plt.savefig(filename)
except (RuntimeError, np.linalg.linalg.LinAlgError, ValueError) as e: except (
RuntimeError,
np.linalg.linalg.LinAlgError,
ValueError,
OverflowError,
) as e:
logger.warning(e) logger.warning(e)
logger.warning("Failed to create dynesty run plot at checkpoint") logger.warning("Failed to create dynesty run plot at checkpoint")
except Exception as e:
logger.warning(
f"Unexpected error {e} in dynesty plotting. "
"Please report at git.ligo.org/lscsoft/bilby/-/issues"
)
finally: finally:
plt.close("all") plt.close("all")
try: try:
filename = f"{self.outdir}/{self.label}_checkpoint_stats.png" filename = f"{self.outdir}/{self.label}_checkpoint_stats.png"
fig, axs = dynesty_stats_plot(self.sampler) fig, _ = dynesty_stats_plot(self.sampler)
fig.tight_layout() fig.tight_layout()
plt.savefig(filename) plt.savefig(filename)
except (RuntimeError, ValueError) as e: except (RuntimeError, ValueError, OverflowError) as e:
logger.warning(e) logger.warning(e)
logger.warning("Failed to create dynesty stats plot at checkpoint") logger.warning("Failed to create dynesty stats plot at checkpoint")
except DynestySetupError: except DynestySetupError:
logger.debug("Cannot create Dynesty stats plot with dynamic sampler.") logger.debug("Cannot create Dynesty stats plot with dynamic sampler.")
except Exception as e:
logger.warning(
f"Unexpected error {e} in dynesty plotting. "
"Please report at git.ligo.org/lscsoft/bilby/-/issues"
)
finally: finally:
plt.close("all") plt.close("all")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment