__init__.py

from .DCGAN import DCGAN_generator
from .DCGAN import DCGAN_discriminator

from .ConditionalGAN import ConditionalGAN_generator
from .ConditionalGAN import ConditionalGAN_discriminator

from .WassersteinCGAN import WCGAN_generator
from .WassersteinCGAN import WCGAN_discriminator

from .DRGAN import DRGAN_encoder
from .DRGAN import DRGAN_decoder
from .DRGAN import DRGAN_discriminator

from .DRGANOriginal import DRGANOriginal_encoder
from .DRGANOriginal import DRGANOriginal_decoder
from .DRGANOriginal import DRGANOriginal_discriminator

from .DCGAN import weights_init

# gets sphinx autodoc done right - don't remove it
__all__ = [_ for _ in dir() if not _.startswith('_')]