aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/cli/main.py
diff options
context:
space:
mode:
authorDmitrii Agibov <dmitrii.agibov@arm.com>2023-02-20 15:42:33 +0000
committerDmitrii Agibov <dmitrii.agibov@arm.com>2023-02-21 15:52:39 +0000
commit803a91c0723533f62148528a81f9d0411b57438e (patch)
tree953b9bd41aa86edf030e009758280d46b8556fe6 /src/mlia/cli/main.py
parenta01e0e1229e30f9e2ebf3cd921ffe0dbef1cbea9 (diff)
downloadmlia-803a91c0723533f62148528a81f9d0411b57438e.tar.gz
MLIA-813 Change default output directory0.6.0-rc.30.6.0
- Use directory mlia-output as output directory for MLIA - If parameter --output-dir provided then place directory mlia-output under specified path or otherwise create it in the current working directory Change-Id: I298088c4aa8dbe9f35dee69ecb9ff6e9ea3cac0a
Diffstat (limited to 'src/mlia/cli/main.py')
-rw-r--r--src/mlia/cli/main.py16
1 files changed, 10 insertions, 6 deletions
diff --git a/src/mlia/cli/main.py b/src/mlia/cli/main.py
index cc97494..88258d5 100644
--- a/src/mlia/cli/main.py
+++ b/src/mlia/cli/main.py
@@ -160,12 +160,16 @@ def setup_context(
args: argparse.Namespace, context_var_name: str = "ctx"
) -> tuple[ExecutionContext, dict]:
"""Set up context and resolve function parameters."""
- ctx = ExecutionContext(
- verbose="debug" in args and args.debug,
- action_resolver=CLIActionResolver(vars(args)),
- output_format=get_output_format(args),
- output_dir=args.output_dir if "output_dir" in args else None,
- )
+ try:
+ ctx = ExecutionContext(
+ verbose="debug" in args and args.debug,
+ action_resolver=CLIActionResolver(vars(args)),
+ output_format=get_output_format(args),
+ output_dir=args.output_dir if "output_dir" in args else None,
+ )
+ except Exception as err: # pylint: disable=broad-except
+ print(f"Error: {err}", file=sys.stderr)
+ sys.exit(1)
setup_logging(ctx.logs_path, ctx.verbose, ctx.output_format)