Skip to content

Commit a0c9b6e

Browse files
authored
Create dummy measurements.json file if not present (#216)
* create dummy measurements.json if file not present * [Automated Commit] Format Codebase [skip ci] * Update customize.py
1 parent 532dd21 commit a0c9b6e

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

script/generate-mlperf-inference-submission/customize.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -553,9 +553,17 @@ def generate_submission(env, state, inp, submission_division):
553553
shutil.copy(measurements_json_path, destination)
554554

555555
else:
556-
if mode.lower() == "performance":
557-
return {
558-
"return": 1, "error": f"measurements.json missing in both paths: {measurements_json_path} and {os.path.join(result_scenario_path, 'user.conf')}"}
556+
print(
557+
f"Warning: measurements.json file not present, creating a dummy measurements.json in path {measurements_json_path}")
558+
dummy_measurements_data = {
559+
"input_data_types": env['MLC_ML_MODEL_INPUTS_DATA_TYPE'] if env.get('MLC_ML_MODEL_INPUTS_DATA_TYPE') else "TBD",
560+
"retraining": env['MLC_ML_MODEL_RETRAINING'] if env.get('MLC_ML_MODEL_RETRAINING') else "TBD",
561+
"starting_weights_filename": env['MLC_ML_MODEL_STARTING_WEIGHTS_FILENAME'] if env.get('MLC_ML_MODEL_STARTING_WEIGHTS_FILENAME') else "TBD",
562+
"weight_data_types": env['MLC_ML_MODEL_WEIGHTS_DATA_TYPE'] if env.get('MLC_ML_MODEL_WEIGHTS_DATA_TYPE') else "TBD",
563+
"weight_transformations": env['MLC_ML_MODEL_WEIGHT_TRANSFORMATIONS'] if env.get('MLC_ML_MODEL_WEIGHT_TRANSFORMATIONS') else "TBD"
564+
}
565+
with open(measurements_json_path, 'w') as json_file:
566+
json.dump(data, json_file, indent=4)
559567

560568
files = []
561569
readme = False

0 commit comments

Comments
 (0)