fix import in main directory

2019-07-26 11:22:46 +02:00 · 2019-07-26 11:22:46 +02:00 · 5c695d3cea
commit 5c695d3cea
parent 1d125683dd
2 changed files with 2 additions and 3 deletions
--- a/monoloco/init.py
+++ b/monoloco/init.py
@ -2,4 +2,3 @@
 """Open implementation of MonoLoco."""

 __version__ = '0.4.3'
-
--- a/monoloco/run.py
+++ b/monoloco/run.py
@ -100,8 +100,8 @@ def main():
            from .visuals.webcam import webcam
            webcam(args)
        else:
-            from predict import predict
-            predict(args)
+            from . import predict
+            predict.predict(args)

    elif args.command == 'prep':
        if 'nuscenes' in args.dataset: