diff --git a/vtr_flow/scripts/run_vtr_flow.py b/vtr_flow/scripts/run_vtr_flow.py index 118ab030186..3d584274e3f 100755 --- a/vtr_flow/scripts/run_vtr_flow.py +++ b/vtr_flow/scripts/run_vtr_flow.py @@ -10,6 +10,7 @@ import socket from datetime import datetime from collections import OrderedDict +import os # pylint: disable=wrong-import-position, import-error sys.path.insert(0, str(Path(__file__).resolve().parent / "python_libs")) @@ -186,8 +187,8 @@ def vtr_command_argparser(prog=None): house_keeping.add_argument( "-temp_dir", - default=None, - help="Directory to run the flow in (will be created if non-existent).", + default=os.getcwd() + "/temp", + help="Absolute Directory to run the flow in (will be created if non-existent).", ) house_keeping.add_argument("-name", default=None, help="Name for this run to be output.") @@ -521,10 +522,9 @@ def vtr_command_main(arg_list, prog=None): # Load the arguments args, unknown_args = vtr_command_argparser(prog).parse_known_args(arg_list) error_status = "Error" - if args.temp_dir is None: - temp_dir = Path("./temp") - else: - temp_dir = Path(args.temp_dir) + + assert args.temp_dir + temp_dir = Path(args.temp_dir) # Specify how command should be run command_runner = vtr.CommandRunner( track_memory=args.track_memory_usage,