__main__.py: stop breaking automat-visualize

2018-04-10 21:33:50 -07:00 · 2018-04-10 21:33:50 -07:00 · e260369be1
commit e260369be1
parent eb7c04e366
1 changed files with 6 additions and 6 deletions
--- a/src/wormhole/main.py
+++ b/src/wormhole/main.py
@ -1,7 +1,7 @@
 from __future__ import absolute_import, print_function, unicode_literals
-from .cli import cli
-
-if __name__ != "__main__":
-    raise ImportError('this module should not be imported')
-
-cli.wormhole()
+if __name__ == "__main__":
+    from .cli import cli
+    cli.wormhole()
+else:
+    # raise ImportError('this module should not be imported')
+    pass