Type TaskGroup in active_states (#4408)

This was missing a type annotation before, which results in issues when accessing `_states` when compiled. This fixes that issue.
dask · Jan 7, 2021 · 49e19c0 · 49e19c0
1 parent 7fb5d4d
commit 49e19c0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -708,6 +708,7 @@ def active(self):
 
     @property
     def active_states(self):
+        tg: TaskGroup
         return merge_with(sum, [tg._states for tg in self.active])
 
     def __repr__(self):