"""
This type stub file was generated by pyright.
"""

from enum import Enum

"""DGL Distributed Training Infrastructure."""
_WAIT_TIME_SEC = ...
class KVMsgType(Enum):
    """Type of kvstore message
    """
    FINAL = ...
    INIT = ...
    PUSH = ...
    PULL = ...
    PULL_BACK = ...
    BARRIER = ...
    IP_ID = ...
    GET_SHAPE = ...
    GET_SHAPE_BACK = ...


KVStoreMsg = ...
