diff --git a/python/cli/convert/tum.py b/python/cli/convert/tum.py index 7fcb0a2..0a04d43 100644 --- a/python/cli/convert/tum.py +++ b/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): diff --git a/python/cli/sai_cli.py b/python/cli/sai_cli.py index 776f740..1b2c729 100644 --- a/python/cli/sai_cli.py +++ b/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() diff --git a/python/cli/smooth.py b/python/cli/smooth.py index 0600a99..d88b763 100644 --- a/python/cli/smooth.py +++ b/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")