diff --git a/assets/batch_score/components/driver/src/batch_score_oss/utils/v2_output_formatter.py b/assets/batch_score/components/driver/src/batch_score_oss/utils/v2_output_formatter.py index 9ba1ca2ab4..f1314bf30f 100644 --- a/assets/batch_score/components/driver/src/batch_score_oss/utils/v2_output_formatter.py +++ b/assets/batch_score/components/driver/src/batch_score_oss/utils/v2_output_formatter.py @@ -75,9 +75,9 @@ def format_output(self, results: "list[ScoringResult]", batch_size_per_request: def __get_request_id(self, scoring_request: ScoringResult): return ( - scoring_request.response_headers.get("x-request-id", "") + scoring_request.response_headers.get("x-request-id", None) if scoring_request.response_headers is not None - else "" + else None ) def _get_response_obj(self, result: dict):