Update tests/test_properties.py

xarray-contrib · Jan 8, 2025 · caa98b8 · caa98b8
1 parent 0b9fafc
commit caa98b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_properties.py b/tests/test_properties.py
@@ -318,5 +318,5 @@ def test_topk_max_min(data, array):
         actual, _ = groupby_reduce(a, by, func="topk", finalize_kwargs={"k": k})
         # TODO: do numbagg, flox
         # FIXME: this is wrong, remove this compute, add a property test checking dask vs numpy
-        expected, _ = groupby_reduce(dask.compute(a)[0], by, func=npfunc, engine="numpy")
+        expected, _ = groupby_reduce(a, by, func=npfunc, engine="numpy")
         assert_equal(actual, expected[np.newaxis, :])