docstr: All TaskGroups belonging to this branch of span tree
docstr_tok: ["All","TaskGroups","belonging","to","this","branch","of","span","tree"]
code: def traverse_groups(self) -> Iterator[TaskGroup]:
      """All TaskGroups belonging to this branch of span tree"""
      for span in self.traverse_spans():
          yield from span.groups
  
imports: from __future__ import annotations
  import uuid
  import weakref
  from collections import defaultdict
  from collections.abc import Hashable, Iterable, Iterator
  from contextlib import contextmanager
  from itertools import islice
  from typing import TYPE_CHECKING, Any
  import dask.config
  from dask.typing import Key
  from distributed.collections import sum_mappings
  from distributed.itertools import ffill
  from distributed.metrics import time
label_desc: Use image node_id 6 for calling the Span obj's underlying member method code with example usage: obj.traverse_groups() without return types