Skip to content

Commit 996bb1a

Browse files
committed
When running vtr flow, make track_memory_usage look at the value of the argument, also made default value of argument true
1 parent f4cc9a3 commit 996bb1a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vtr_flow/scripts/run_vtr_flow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ def vtr_command_argparser(prog=None):
184184

185185
house_keeping.add_argument(
186186
"-track_memory_usage",
187-
default=False,
187+
default=True,
188188
action="store_true",
189189
dest="track_memory_usage",
190190
help="Track the memory usage for each stage."
@@ -406,7 +406,7 @@ def vtr_command_main(arg_list, prog=None):
406406
temp_dir = Path(args.temp_dir)
407407
# Specify how command should be run
408408
command_runner = vtr.CommandRunner(
409-
track_memory=True,
409+
track_memory=args.track_memory_usage,
410410
max_memory_mb=args.limit_memory_usage,
411411
timeout_sec=args.timeout,
412412
verbose=args.verbose,

0 commit comments

Comments
 (0)