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: