Add support for both script and module local import (#45)

lavie · Jul 8, 2019 · a849120 · a849120
1 parent e2d2dd0
commit a849120
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/runlike/runlike.py b/runlike/runlike.py
@@ -1,8 +1,11 @@
 #!/usr/bin/env python
 
 import click
-from inspector import Inspector
 
+try:
+    from .inspector import Inspector
+except ValueError:
+    from inspector import Inspector
 
 
 @click.command(