aaaa!! dont panic. I broke imports quick fix

mfinzi · Apr 20, 2021 · 9667679 · 9667679
1 parent eb664d1
commit 9667679
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/emlp/nn/__init__.py b/emlp/nn/__init__.py
@@ -1,2 +1,6 @@
-
-from .objax import *
+import importlib
+import pkgutil
+__all__ = []
+module = importlib.import_module('.'+'objax',package=__name__)
+globals().update({k: getattr(module, k) for k in module.__all__})
+__all__ += module.__all__