Another bug fix with the ML HPC run.

Author: KriFos1

ensemble/ensemble.py

@@ -611,7 +611,7 @@ def calc_ml_prediction(self, input_state=None):
                         en_pred[list_crash[indx]] = deepcopy(en_pred[el])
 
                 # Convert ensemble specific result into pred_data, and filter for NONE data
-                ml_pred_data.append([{typ: np.concatenate(tuple((el[ind][typ][:, np.newaxis]) for el in en_pred), axis=1)
+                ml_pred_data.append([{typ: np.concatenate(tuple((np.atleast_2d(el[ind][typ]).T) for el in en_pred), axis=1)
                                       if any(elem is not None for elem in tuple((el[ind][typ]) for el in en_pred))
                                       else None for typ in en_pred[0][0].keys()} for ind in range(len(en_pred[0]))])