--- a
+++ b/ext/lab2im/__init__.py
@@ -0,0 +1,6 @@
+from . import edit_tensors
+from . import edit_volumes
+from . import image_generator
+from . import lab2im_model
+from . import layers
+from . import utils