Fix DataArray.__dask_scheduler__ to threaded scheduler (#1760)

Author: mrocklin

doc/whats-new.rst

@@ -35,6 +35,8 @@ Bug fixes
 
 .. _whats-new.0.10.0:
 
+- Properly point DataArray.__dask_scheduler__ to dask.threaded.get
+
 v0.10.0 (20 November 2017)
 --------------------------
 

xarray/core/dataarray.py

@@ -594,7 +594,7 @@ def __dask_optimize__(self):
 
     @property
     def __dask_scheduler__(self):
-        return self._to_temp_dataset().__dask_optimize__
+        return self._to_temp_dataset().__dask_scheduler__
 
     def __dask_postcompute__(self):
         func, args = self._to_temp_dataset().__dask_postcompute__()

xarray/tests/test_dask.py

@@ -830,3 +830,16 @@ def test_dataarray_with_dask_coords():
     assert not dask.is_dask_collection(array2)
 
     assert all(isinstance(v._variable.data, np.ndarray) for v in array2.coords.values())
+
+
+def test_basic_compute():
+    ds = Dataset({'foo': ('x', range(5)),
+                  'bar': ('x', range(5))}).chunk({'x': 2})
+    for get in [dask.threaded.get,
+                dask.multiprocessing.get,
+                dask.local.get_sync,
+                None]:
+        with dask.set_options(get=get):
+            ds.compute()
+            ds.foo.compute()
+            ds.foo.variable.compute()