From b8e7cd479a085b6cde87842658d887f582d5bd4c Mon Sep 17 00:00:00 2001 From: anandhu-eng Date: Tue, 1 Apr 2025 10:57:37 +0530 Subject: [PATCH] handle submission base directory properly --- .../customize.py | 19 +++++++------------ 1 file changed, 7 insertions(+), 12 deletions(-) diff --git a/script/generate-mlperf-inference-submission/customize.py b/script/generate-mlperf-inference-submission/customize.py index a10ce68a5..cbe8b430c 100644 --- a/script/generate-mlperf-inference-submission/customize.py +++ b/script/generate-mlperf-inference-submission/customize.py @@ -72,20 +72,15 @@ def generate_submission(env, state, inp, submission_division): sys.path.append(submission_checker_dir) if env.get('MLC_MLPERF_INFERENCE_SUBMISSION_DIR', '') == '': - user_home = str(Path.home()) - env['MLC_MLPERF_INFERENCE_SUBMISSION_DIR'] = os.path.join( - user_home, "mlperf_submission") + if env.get('MLC_MLPERF_INFERENCE_SUBMISSION_BASE_DIR', '') == '': + user_home = str(Path.home()) + env['MLC_MLPERF_INFERENCE_SUBMISSION_DIR'] = os.path.join( + user_home, "mlperf_submission") + else: + env['MLC_MLPERF_INFERENCE_SUBMISSION_DIR'] = os.path.join( + env['MLC_MLPERF_INFERENCE_SUBMISSION_BASE_DIR'], "mlperf_submission") submission_dir = env.get('MLC_MLPERF_INFERENCE_SUBMISSION_DIR', '') - if submission_dir == '': - submission_base_dir = env.get( - 'MLC_MLPERF_INFERENCE_SUBMISSION_BASE_DIR', '') - if submission_base_dir == '': - return {'return': 1, 'error': f"Both MLC_MLPERF_INFERENCE_SUBMISSION_DIR and MLC_MLPERF_INFERENCE_SUBMISSION_BASE_DIR can not be empty!"} - else: - submission_dir = os.path.join( - submission_base_dir, "mlperf_inference_submission") - env['MLC_MLPERF_INFERENCE_SUBMISSION_DIR'] = submission_dir if env.get('MLC_MLPERF_CLEAN_SUBMISSION_DIR', '') != '': print('=================================================')