fix flakes

Azure · Sep 18, 2024 · b9d4d51 · b9d4d51
1 parent 9267c87
commit b9d4d51
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/...atch_score/components/driver/src/batch_score_oss/common/scoring/scoring_client_factory.py b/...atch_score/components/driver/src/batch_score_oss/common/scoring/scoring_client_factory.py
@@ -52,16 +52,15 @@ def setup_scoring_client(
         if endpoint_type is EndpointType.Null:
             get_logger().info("Using null endpoint scoring client.")
             return _setup_null_endpoint_scoring_client(
-            configuration=configuration,
-            tally_handler=tally_handler)
+                configuration=configuration,
+                tally_handler=tally_handler)
 
         get_logger().info("Using MIR scoring client.")
         return _setup_mir_scoring_client(
             configuration=configuration,
             header_provider=header_provider,
             tally_handler=tally_handler)
 
-
     def _get_additional_headers(self, configuration):
         if configuration.additional_headers is None:
             headers = {}

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
@@ -75,7 +75,7 @@ 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", None)
+            scoring_request.response_headers.get("x-request-id", "")
             if scoring_request.response_headers is not None
             else None
         )