Skip to content

Commit 5137820

Browse files
committed
minor test script cleanup
1 parent 0a3219d commit 5137820

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

megadetector/utils/md_tests.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1272,13 +1272,14 @@ def run_cli_tests(options):
12721272
print('\n** Running MD on a folder (with image queue and preprocessing) (CLI) **\n')
12731273

12741274
cmd = base_cmd + ' --use_image_queue --preprocess_on_image_queue'
1275-
inference_output_file_queue = insert_before_extension(inference_output_file,'preprocess_queue')
1276-
cmd = cmd.replace(inference_output_file,inference_output_file_queue)
1275+
inference_output_file_preprocess_queue = \
1276+
insert_before_extension(inference_output_file,'preprocess_queue')
1277+
cmd = cmd.replace(inference_output_file,inference_output_file_preprocess_queue)
12771278
cmd += ' --detector_options {}'.format(dict_to_kvp_list(options.detector_options))
12781279
cmd_results = execute_and_print(cmd)
12791280

12801281
assert output_files_are_identical(fn1=inference_output_file,
1281-
fn2=inference_output_file_queue,
1282+
fn2=inference_output_file_preprocess_queue,
12821283
verbose=True)
12831284

12841285
## Run again on multiple cores, make sure the results are the same

0 commit comments

Comments
 (0)