class_docstr: registry for parameters in flight
code: def __setitem__(
      self, param: Parameter, handle: AllGatherCoalescedHandle
  ) -> None:
      if param in self.data:
          raise RuntimeError(
              f"{param.ds_summary()} already in registry"
          )
      if param.ds_status != ZeroParamStatus.INFLIGHT:
          raise RuntimeError(
              f"attempted to add non-inflight parameter to registry {param.ds_summary()}"
          )
      self.data[param] = handle
  
imports: from dataclasses import dataclass
  import collections
  from collections import UserDict
  from typing import Deque, Set
  from deepspeed import comm
  from deepspeed.utils.logging import logger
  from deepspeed.runtime.zero.offload_config import OffloadDeviceEnum
  from deepspeed.runtime.zero.partition_parameters import *
  from deepspeed.runtime.zero.partitioned_param_profiler import PartitionedParameterProfiler
  from deepspeed.runtime.swap_tensor.partitioned_param_swapper import PartitionedParamStatus
  from deepspeed.utils.debug import debug_module2name_id, debug_param2name_id
  from deepspeed.accelerator import get_accelerator
  import logging
label_desc: Use image node_id 1 for calling the InflightParamRegistry obj's underlying member method code with example usage: obj.__setitem__(param, handle) without return types