Merge pull request #1224 from PyAutoLabs/feature/autofit-smoke-cleanup

Guard corner_cornerpy against samples < dims

Author: Jammy2211

autofit/non_linear/plot/samples_plotters.py

@@ -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"]
@@ -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,
     )