# This file is never automatically imported within PyTorch so it is ok to
# always warn here
import warnings

warnings.warn("The `torch.nn.utils._stateless` code is deprecated now that "
              "it is publicly available. Please use `torch.nn.utils.stateless "
              "instead.", DeprecationWarning)

# Import * wouldn't work as most things are private and thus wouldn't be imported
# here.
from torch.nn.utils.stateless import functional_call  # noqa: F401
from torch.nn.utils.stateless import _apply_func_submodules, _change_class  # noqa: F401
# This one used to look public but should actually be private. This was fixed when making the module
# public and is kept here for BC
from torch.nn.utils.stateless import _reparametrize_module as reparametrize_module  # noqa: F401