# Keep old package for BC purposes, this file should be removed once
# everything moves to the `torch.distributed.checkpoint` package.
import sys
import torch
import warnings

from torch.distributed.checkpoint import *  # noqa: F403
warnings.warn(
    "torch.distributed._shard.checkpoint will be deprecated, use torch.distributed.checkpoint instead",
    DeprecationWarning
)
sys.modules['torch.distributed._shard.checkpoint'] = torch.distributed.checkpoint