From f39f2ddbe61cd429215711b67392f12ecefd74ac Mon Sep 17 00:00:00 2001 From: dcherian Date: Sat, 3 Aug 2019 17:44:39 -0600 Subject: [PATCH 1/2] Call darray.compute() in plot() --- xarray/plot/plot.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/xarray/plot/plot.py b/xarray/plot/plot.py index 9d0bf671dda..1c6719ad00d 100644 --- a/xarray/plot/plot.py +++ b/xarray/plot/plot.py @@ -10,6 +10,7 @@ import numpy as np import pandas as pd +from ..core.pycompat import dask_array_type from .facetgrid import _easy_facetgrid from .utils import ( _add_colorbar, _ensure_plottable, _infer_interval_breaks, _infer_xy_labels, @@ -141,6 +142,9 @@ def plot(darray, row=None, col=None, col_wrap=None, ax=None, hue=None, """ darray = darray.squeeze() + if isinstance(darray.data, dask_array_type): + darray = darray.compute() + plot_dims = set(darray.dims) plot_dims.discard(row) plot_dims.discard(col) From 0000cb02032d9bd0715fe8d7791061697aeec7b1 Mon Sep 17 00:00:00 2001 From: dcherian Date: Sun, 4 Aug 2019 16:33:27 -0600 Subject: [PATCH 2/2] review. --- xarray/plot/plot.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/xarray/plot/plot.py b/xarray/plot/plot.py index 1c6719ad00d..d0003b702df 100644 --- a/xarray/plot/plot.py +++ b/xarray/plot/plot.py @@ -10,7 +10,6 @@ import numpy as np import pandas as pd -from ..core.pycompat import dask_array_type from .facetgrid import _easy_facetgrid from .utils import ( _add_colorbar, _ensure_plottable, _infer_interval_breaks, _infer_xy_labels, @@ -140,10 +139,7 @@ def plot(darray, row=None, col=None, col_wrap=None, ax=None, hue=None, Additional keyword arguments to matplotlib """ - darray = darray.squeeze() - - if isinstance(darray.data, dask_array_type): - darray = darray.compute() + darray = darray.squeeze().compute() plot_dims = set(darray.dims) plot_dims.discard(row)