from torch import nn

__all__ = ["PublicClass", "PublicSubclass", "MyModule"]


class PublicClass:
    __module__ = "my_package"


class PublicSubclass(PublicClass):
    """:class:`my_package.PublicClass`"""
    pass


class MyModule(nn.Module):
    """:class:`torch.nn.Module`"""
    pass