class_docstr: Keep track of all keys, grouped by key_split
class_docstr_tok: ["Keep","track","of","all","keys",",","grouped","by","key_split"]
code: def restart(self, scheduler):
      self.all.clear()
      self.state.clear()
  
imports: from __future__ import annotations
  import asyncio
  import logging
  import warnings
  from collections import defaultdict
  from timeit import default_timer
  from typing import ClassVar
  from tlz import groupby, valmap
  from dask.base import tokenize
  from dask.utils import key_split
  from distributed.diagnostics.plugin import SchedulerPlugin
  from distributed.metrics import time
label_desc: Use image node_id 3 for calling the AllProgress obj's underlying member method code with example usage: obj.restart(scheduler) without return types