shared/datastore/__init__.py

@@ -1,8 +1,10 @@
-from .connect_minio import connect_minio
-from .delete import delete
-from .get import get
-from .get_image import get_image
-from .get_model import get_model
-from .put import put
-from .put_image import put_image
-from .put_model import put_model
+from .src import (
+    connect_minio,
+    delete,
+    get,
+    get_image,
+    get_model,
+    put,
+    put_image,
+    put_model,
+)

shared/datastore/src/__init__.py

@@ -0,0 +1,8 @@
+from .connect_minio import connect_minio
+from .delete import delete
+from .get import get
+from .get_image import get_image
+from .get_model import get_model
+from .put import put
+from .put_image import put_image
+from .put_model import put_model