mfinzi/pristine-ml

# from .CoordConv import CoordConv
# from .blocks import conv2d,ConvBNrelu,FcBNrelu,ResBlock, DenseBlock,ConcatResBlock,ODEBlock,RNNBlock
# from .blocks import FiLMResBlock,ConcatBottleBlock
# #from .denseblocks import DenseLayer, DenseBlock, TransitionUp,TransitionDown,Bottleneck
# __all__ = ['CoordConv','conv2d','ConvBNrelu','FcBNrelu','ResBlock','DenseBlock','ConcatResBlock',
#             'ODEBlock','RNNBlock','FiLMResBlock','ConcatBottleBlock']
#     #'DenseLayer', 'DenseBlock', 'TransitionUp','TransitionDown','Bottleneck']

import importlib
import pkgutil
__all__ = []
for loader, module_name, is_pkg in  pkgutil.walk_packages(__path__):
    module = importlib.import_module('.'+module_name,package=__name__)
    try: 
        globals().update({k: getattr(module, k) for k in module.__all__})
        __all__ += module.__all__
    except AttributeError: continue