From 4f3969b797071651ec6d9a84a1d034605f4afd0a Mon Sep 17 00:00:00 2001 From: Matthew Rocklin Date: Sun, 27 May 2018 11:17:09 -0400 Subject: [PATCH] Fix a dataframe test to match dask/dask change --- distributed/tests/test_collections.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/distributed/tests/test_collections.py b/distributed/tests/test_collections.py index 46aa8495332..370dbecfab9 100644 --- a/distributed/tests/test_collections.py +++ b/distributed/tests/test_collections.py @@ -161,7 +161,7 @@ def test_dataframe_groupby_tasks(loop): assert_equal(a, b.compute(scheduler='sync').sort_index()) assert not any('partd' in k[0] for k in b.dask) - with pytest.raises(NotImplementedError): + with pytest.raises((NotImplementedError, ValueError)): ddf.groupby(ddf[['A', 'B']]).apply(len, meta=int) a = df.groupby(['A', 'B']).apply(len)