From f6dbe11e9233df53ae030b847f0669acb9d6b681 Mon Sep 17 00:00:00 2001 From: Philipp Rudiger Date: Fri, 2 Jun 2017 12:06:45 +0100 Subject: [PATCH] Allowed using xarray with dask arrays --- holoviews/core/data/xarray.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/holoviews/core/data/xarray.py b/holoviews/core/data/xarray.py index 9fbeb928bc..e0a5a41d0a 100644 --- a/holoviews/core/data/xarray.py +++ b/holoviews/core/data/xarray.py @@ -5,6 +5,11 @@ import numpy as np import xarray as xr +try: + import dask +except ImportError: + dask = None + from .. import util from ..dimension import Dimension from ..ndmapping import NdMapping, item_check, sorted_context @@ -156,6 +161,8 @@ def values(cls, dataset, dim, expanded=True, flat=True): data = dataset.data[dim.name].data if dim in dataset.vdims: coord_dims = dataset.data[dim.name].dims + if dask and isinstance(data, dask.array.Array): + data = data.compute() data = cls.canonicalize(dataset, data, coord_dims=coord_dims) return data.T.flatten() if flat else data elif expanded: