-
-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Closed
Labels
Description
As of
- dask 2.1.0
- distributed 2.1.0
- xarray 0.12.1 or git head (didn't try older versions):
>>> import xarray
>>> import distributed
>>> client = distributed.Client(set_as_default=False)
>>> ds = xarray.Dataset({'d': ('x', [1, 2])}).chunk(1)
>>> client.compute(ds).result()
<xarray.Dataset>
Dimensions: (x: 2)
Dimensions without coordinates: x
Data variables:
d (x) int64 1 2
>>> client.compute(ds.d).result()
distributed.worker - WARNING - Compute Failed
Function: _dask_finalize
args: ([[array([1]), array([2])]], <function Dataset._dask_postcompute at 0x316a1db70>, ([(True, <this-array>, (<function Variable._dask_finalize at 0x3168f7f28>, (<function finalize at 0x1166bb8c8>, (), ('x',), OrderedDict(), None)))], set(), {'x': 2}, None, None, None, None), 'd')
kwargs: {}
Exception: KeyError(<this-array>)
---------------------------------------------------------------------------
KeyError Traceback (most recent call last)
<ipython-input-8-2dbfe1b2ff17> in <module>
----> 1 client.compute(ds.d).result()
/anaconda3/lib/python3.7/site-packages/distributed/client.py in result(self, timeout)
226 result = self.client.sync(self._result, callback_timeout=timeout, raiseit=False)
227 if self.status == "error":
--> 228 six.reraise(*result)
229 elif self.status == "cancelled":
230 raise result
/anaconda3/lib/python3.7/site-packages/six.py in reraise(tp, value, tb)
690 value = tp()
691 if value.__traceback__ is not tb:
--> 692 raise value.with_traceback(tb)
693 raise value
694 finally:
~/PycharmProjects/xarray/xarray/core/dataarray.py in _dask_finalize()
706 def _dask_finalize(results, func, args, name):
707 ds = func(results, *args)
--> 708 variable = ds._variables.pop(_THIS_ARRAY)
709 coords = ds._variables
710 return DataArray(variable, coords, name=name, fastpath=True)