Merge pull request #76 from GeoOcean/75-mdaplot-two-vars

[JTH] fix bug when 2 varables and just 1 ax

Author: tausiaj

bluemath_tk/datamining/_base_datamining.py

@@ -5,6 +5,7 @@
 import pandas as pd
 import xarray as xr
 from matplotlib import pyplot as plt
+from matplotlib.axes import Axes
 
 from ..core.models import BlueMathModel
 from ..core.plotting.base_plotting import DefaultStaticPlotting
@@ -328,6 +329,8 @@ def plot_data_as_clusters(
             sharex=False,
             sharey=False,
         )
+        if isinstance(axes, Axes):
+            axes = np.array([[axes]])
 
         # Gets colors for clusters and append to each nearest centroid
         colors_for_clusters = default_static_plot.get_list_of_colors_for_colormap(