fix import in main directory

This commit is contained in:
lorenzo 2019-07-26 11:22:46 +02:00
parent 1d125683dd
commit 5c695d3cea
2 changed files with 2 additions and 3 deletions

View File

@ -2,4 +2,3 @@
"""Open implementation of MonoLoco.""" """Open implementation of MonoLoco."""
__version__ = '0.4.3' __version__ = '0.4.3'

View File

@ -100,8 +100,8 @@ def main():
from .visuals.webcam import webcam from .visuals.webcam import webcam
webcam(args) webcam(args)
else: else:
from predict import predict from . import predict
predict(args) predict.predict(args)
elif args.command == 'prep': elif args.command == 'prep':
if 'nuscenes' in args.dataset: if 'nuscenes' in args.dataset: