diff --git a/src/gasflux/interpolation.py b/src/gasflux/interpolation.py index 229a3f4..37d7ca0 100644 --- a/src/gasflux/interpolation.py +++ b/src/gasflux/interpolation.py @@ -99,11 +99,6 @@ def ordinary_kriging( # np.nan_to_num(error_1s, copy=False, nan=0) volume_error = simpsonintegrate(error_1s, x_cell_size, y_cell_size) - # Plots disabled - contour_plot = None - grid_plot = None - semivariogram_plot = None - output_text = ( f"The emissions flux of {gas.upper()} is {volume:.3f}kgh⁻¹; " f"the cut and fill volumes of the grid are {volumepos:.3f} and {volumeneg:.3f}kgh⁻¹. " @@ -123,6 +118,20 @@ def ordinary_kriging( "volume_error": volume_error, } + # Generate plots (must be after krig_variables is defined) + try: + contour_plot = plotting.contour_krig(krig_variables) + except Exception: + contour_plot = None + try: + grid_plot = plotting.heatmap_krig(krig_variables) + except Exception: + grid_plot = None + try: + semivariogram_plot = plotting.semivariogram_plot(semivariogram, gas) + except Exception: + semivariogram_plot = None + return krig_variables, output_text, contour_plot, grid_plot, semivariogram_plot