Fix use from inside the SDK Python package

Author: oseiskar

python/cli/process/process.py

@@ -474,7 +474,7 @@ def detect_device_preset(input_dir):
         config['stereoPointCloudStride'] = 30
 
     if args.preview3d:
-        from visualization.visualizer import Visualizer
+        from spectacularAI.cli.visualization.visualizer import Visualizer
         visualizer = Visualizer()
 
     with open(tmp_input + "/vio_config.yaml", 'wt') as f:

python/cli/record/oak.py

@@ -228,7 +228,7 @@ def open_gray_video(name):
     if args.no_preview:
         plotter = None
     else:
-        from visualization.vio_visu import make_plotter
+        from spectacularAI.cli.visualization.vio_visu import make_plotter
         import matplotlib.pyplot as plt
         plotter, anim = make_plotter()
 

python/cli/sai_cli.py

@@ -1,7 +1,7 @@
 import argparse
 
-from process.process import define_subparser as process_define_subparser
-from record.record import define_subparser as record_define_subparser
+from .process.process import define_subparser as process_define_subparser
+from .record.record import define_subparser as record_define_subparser
 
 def parse_args():
     parser = argparse.ArgumentParser(description='Spectacular AI command line tool')
@@ -10,6 +10,9 @@ def parse_args():
     record_define_subparser(subparsers)
     return parser.parse_args()
 
-if __name__ == '__main__':
+def main():
     args = parse_args()
     args.func(args)
+
+if __name__ == '__main__':
+    main()