Small improvements to sai-cli tool

python/cli/convert/tum.py

@@ -11,7 +11,6 @@
 import os
 from pathlib import Path
 import subprocess
-import yaml
 from contextlib import contextmanager
 import shutil
 import tempfile
@@ -86,6 +85,7 @@ def maybe_extract_tar_or_zip(path):
                 print(f"Warning: Failed to delete temporary directory '{temp_dir}': {e}")
 
 def get_calibration(input_dir, stereo):
+    import yaml
     calibration = { "cameras": [] }
 
     def convert_distortion(model, coeffs):

python/cli/sai_cli.py

@@ -19,8 +19,14 @@ def parse_args():
     return parser.parse_args()
 
 def main():
-    args = parse_args()
-    args.func(args)
+    try:
+        args = parse_args()
+        args.func(args)
+    except ImportError as e:
+        print(f"\n[ImportError] {e}")
+        print("Try installing dependencies with")
+        print("    pip install spectacularAI[full]\n")
+        raise
 
 if __name__ == '__main__':
     main()

python/cli/smooth.py

@@ -2,13 +2,7 @@
 Post-process a session and generate a smoothed trajectory with all frames
 """
 import json
-
-try:
-    from process.process import parse_input_dir, auto_config
-except ImportError:
-    # hacky: The following mechanism allows using this both as a stand-alone
-    # script and as a subcommand in sai-cli.
-    from .process.process import parse_input_dir, auto_config
+from .process.process import parse_input_dir, auto_config
 
 def define_args(parser):
     parser.add_argument("input", help="Path to folder with session to process")