--- a
+++ b/dosma/models/__init__.py
@@ -0,0 +1,9 @@
+from dosma.models import oaiunet2d, stanford_qdess, util
+from dosma.models.oaiunet2d import *  # noqa
+from dosma.models.stanford_qdess import *  # noqa: F401, F403
+from dosma.models.util import *  # noqa
+
+__all__ = []
+__all__.extend(util.__all__)
+__all__.extend(oaiunet2d.__all__)
+__all__.extend(stanford_qdess.__all__)