diff --git a/automation/script/module.py b/automation/script/module.py index 8e7556eb4..ac531da64 100644 --- a/automation/script/module.py +++ b/automation/script/module.py @@ -273,31 +273,17 @@ def _run(self, i): # Useful to set up default flags such as the name of virtual enviroment extra_cli = os.environ.get('MLC_SCRIPT_EXTRA_CMD', '').strip() if extra_cli != '': - from cmind import cli - r = cli.parse(extra_cli) + r = convert_args_to_dictionary(extra_cli) if r['return'] > 0: return r - mlc_input = r['mlc_input'] + mlc_input = r['args_dict'] utils.merge_dicts({'dict1': i, 'dict2': mlc_input, 'append_lists': True, 'append_unique': True}) - # Check if has extra tags as a second artifact - # Example: cmr . "_python _tiny" - - parsed_artifacts = i.get('parsed_artifacts', []) - if len(parsed_artifacts) > 0: - extra_tags = parsed_artifacts[0][0][0] - if ' ' in extra_tags or ',' in extra_tags: - # Add tags - x = i.get('tags', '') - if x != '': - x += ',' - i['tags'] = x + extra_tags.replace(' ', ',') - # Recursion spaces needed to format log and print recursion_spaces = i.get('recursion_spaces', '') # Caching selections to avoid asking users again diff --git a/script/app-mlperf-inference/customize.py b/script/app-mlperf-inference/customize.py index f5e3619d0..4248fe894 100644 --- a/script/app-mlperf-inference/customize.py +++ b/script/app-mlperf-inference/customize.py @@ -239,7 +239,7 @@ def postprocess(i): sut_config[model_full_name][scenario][metric] = value print( - f"SUT: {sut_name}, model: {model_full_name}, scenario: {scenario}, {metric} updated as {value}") + f"SUT: {sut_name}, model: {model_full_name}, scenario: {scenario}, {metric} (mean value) updated as {value}") print(f"New config stored in {sut_config_path}") with open(sut_config_path, "w") as f: yaml.dump(sut_config, f)