Skip to content

Commit

Permalink
Enable default values to required env keys
Browse files Browse the repository at this point in the history
  • Loading branch information
anandhu-eng authored Oct 30, 2024
1 parent dd40f0a commit 9e3392b
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions script/generate-mlperf-inference-submission/customize.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ def fill_from_json(file_path, keys, sut_info):
with open(file_path, 'r') as f:
data = json.load(f)
for key in keys:
if key in data and sut_info[key] is None:
if key in data and (sut_info[key] is None or sut_info[key] == "default"):
sut_info[key] = data[key]
elif key in data and sut_info[key] != data[key]:
return -1 # error saying there is a mismatch in the value of a key
Expand Down Expand Up @@ -149,7 +149,8 @@ def generate_submission(i):
"implementation": None,
"device": None,
"framework": None,
"run_config": None
"framework_version": "default",
"run_config": "default"
} # variable to store the system meta

model_mapping_combined = {} # to store all the model mapping related to an SUT
Expand Down

0 comments on commit 9e3392b

Please sign in to comment.