Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 15 additions & 1 deletion autofit/non_linear/plot/samples_plotters.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,26 @@
import logging

import numpy as np

from autoconf import conf

from autofit.non_linear.plot.plot_util import skip_in_test_mode, output_figure

logger = logging.getLogger(__name__)


@skip_in_test_mode
def corner_cornerpy(samples, path=None, filename="corner", format="show", **kwargs):
data = np.asarray(samples.parameter_lists)
if data.ndim < 2 or data.shape[0] <= data.shape[1]:
logger.info(
"corner_cornerpy: skipping corner plot, only %s sample(s) for %s parameter(s) "
"(e.g. PYAUTO_TEST_MODE bypass or an early-iteration update).",
data.shape[0] if data.ndim >= 1 else 0,
data.shape[1] if data.ndim >= 2 else 0,
)
return

import matplotlib.pylab as pylab

config_dict = conf.instance["visualize"]["plots_settings"]["corner_cornerpy"]
Expand All @@ -17,7 +31,7 @@ def corner_cornerpy(samples, path=None, filename="corner", format="show", **kwar
import corner

corner.corner(
data=np.asarray(samples.parameter_lists),
data=data,
weight_list=samples.weight_list,
labels=samples.model.parameter_labels_with_superscripts_latex,
)
Expand Down
Loading