chore: add Union import

ZettaAI · Aug 1, 2023 · 02e3fc4 · 02e3fc4
1 parent 37f156a
commit 02e3fc4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/zetta_utils/__init__.py b/zetta_utils/__init__.py
@@ -1,5 +1,6 @@
 # pylint: disable=unused-import, import-outside-toplevel
 """Zetta AI Computational Connectomics Toolkit."""
+from typing import Union
 
 from . import log, typing, parsing, builder, common
 from . import geometry, distributions, layer, ng
@@ -11,6 +12,8 @@ def load_all_modules():
     load_inference_modules()
     load_training_modules()
     from . import mazepa_addons
+
+    # from . import message_queues
     from . import cloud_management