non underscore group

dask · May 6, 2021 · 03210bc · 03210bc
1 parent 1376706
commit 03210bc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/distributed/tests/test_scheduler.py b/distributed/tests/test_scheduler.py
@@ -1881,13 +1881,13 @@ async def test_task_groups(c, s, a, b):
     assert tg.nbytes_total == tp.nbytes_total
     # It should map down to individual tasks
     assert tg.nbytes_total == sum(
-        [ts.get_nbytes() for ts in s.tasks.values() if ts._group is tg]
+        [ts.get_nbytes() for ts in s.tasks.values() if ts.group is tg]
     )
     in_memory_ts = sum(
         [
             ts.get_nbytes()
             for ts in s.tasks.values()
-            if ts._group is tg and ts.state == "memory"
+            if ts.group is tg and ts.state == "memory"
         ]
     )
     assert tg.nbytes_in_memory == in_memory_ts