m

ZettaAI · Nov 30, 2023 · a640a9d · a640a9d
1 parent a113124
commit a640a9d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/zetta_utils/__init__.py b/zetta_utils/__init__.py
@@ -30,12 +30,14 @@ def try_load_train_inference():  # pragma: no cover
     except ImportError:
         ...
 
-def try_load_submodules(): # pragma: no cover 
+
+def try_load_submodules():  # pragma: no cover
     try:
         from . import alignment
     except ImportError:
         ...
 
+
 def load_inference_modules():
     from . import (
         augmentations,
@@ -50,6 +52,7 @@ def load_inference_modules():
     from .layer import volumetric
     from .layer.volumetric import cloudvol
     from .message_queues import sqs
+
     try_load_submodules()
 
 
@@ -65,6 +68,7 @@ def load_training_modules():
     )
     from .layer import volumetric
     from .layer.volumetric import cloudvol
+
     try_load_submodules()