Skip to content
Snippets Groups Projects
Commit fedaf166 authored by Yifan Zhao's avatar Yifan Zhao
Browse files

Fixed interactive arguments of hpvm installer

parent 121abda4
No related branches found
No related tags found
No related merge requests found
......@@ -120,11 +120,15 @@ def prompt_args():
"Build target: ", parse_targets, "Input shouldn't contain space"
)
download_weights = input_with_check(
"Download DNN weights (recommended)? [y/n]: ", parse_yn, "Please enter y or n"
)
run_tests = input_with_check(
"Build and run tests? [y/n]: ", parse_yn, "Please enter y or n"
)
return Namespace(
no_build=not auto_build, parallel=threads, targets=targets, run_tests=run_tests
no_build=not auto_build, parallel=threads, targets=targets, run_tests=run_tests,
no_params=not download_weights
)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment