Fix groupby test after meta requirements got stricter in
@@ -168,7 +168,7 @@ def test_dataframe_groupby_tasks(client):
for ind in [lambda x: "A", lambda x: x.A]:
a = df.groupby(ind(df)).apply(len)
- b = ddf.groupby(ind(ddf)).apply(len, meta=int)
+ b = ddf.groupby(ind(ddf)).apply(len, meta=(None, int))
assert_equal(a, b.compute(scheduler="sync").sort_index())
assert not any("partd" in k[0] for k in b.dask)
@@ -181,7 +181,7 @@ def test_dataframe_groupby_tasks(client):
ddf.groupby(ddf[["A", "B"]]).apply(len, meta=int)
a = df.groupby(["A", "B"]).apply(len)
- b = ddf.groupby(["A", "B"]).apply(len, meta=int)
+ b = ddf.groupby(["A", "B"]).apply(len, meta=(None, int))
assert_equal(a, b.compute(scheduler="sync").sort_index())
