diff --git a/GPy/plotting/matplot_dep/plot_definitions.py b/GPy/plotting/matplot_dep/plot_definitions.py index 34f32ac5f..dc7ea87f2 100644 --- a/GPy/plotting/matplot_dep/plot_definitions.py +++ b/GPy/plotting/matplot_dep/plot_definitions.py @@ -241,8 +241,8 @@ def pairwise(iterable): except ImportError: from matplotlib.mlab import contiguous_regions # Handle united data, such as dates - ax._process_unit_info(xdata=X, ydata=y1) - ax._process_unit_info(ydata=y2) + ax._process_unit_info([("x", X), ("y", y1)], convert=False) + ax._process_unit_info([("y", y2)], convert=False) # Convert the arrays so we can work with them from numpy import ma x = ma.masked_invalid(ax.convert_xunits(X))