codebase/models/shared/__init__.py

import glob
from os.path import dirname, basename, isfile
files = glob.glob(dirname(__file__) + '/*.py')
__all__ = [basename(f)[:-3] for f in files if isfile(f) and not f.endswith('__init__.py')]
from .mask import *
from .causal_masked_flow import MultivariateCausalFlow, PriorMultivariateCausalFlow, CausalAffineAutoregFlow