chore(components): [endpoint_batch_predict] add qms_override paramete…

…r data type conversion PiperOrigin-RevId: 578698352
kubeflow · Nov 2, 2023 · d9c3b86 · d9c3b86
1 parent ed9a5ab
commit d9c3b86
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/..._pipeline_components/_implementation/model_evaluation/endpoint_batch_predict/component.py b/..._pipeline_components/_implementation/model_evaluation/endpoint_batch_predict/component.py
@@ -143,7 +143,7 @@ def evaluation_llm_endpoint_batch_predict_pipeline_graph_component(
     model_parameters: Optional[Dict[str, Union[int, float]]] = {},
     endpoint_id: Optional[str] = '',
     publisher_model: Optional[str] = '',
-    qms_override: Optional[str] = None,
+    qms_override: Optional[Dict[str, Union[int, float]]] = None,
     display_name: str = 'endpoint_batch_predict',
     machine_type: str = 'e2-highmem-16',
     service_account: str = '',
@@ -214,7 +214,7 @@ def evaluation_llm_endpoint_batch_predict_pipeline_graph_component(
       ).output,
       endpoint_id=endpoint_id,
       publisher_model=publisher_model,
-      qms_override=qms_override,
+      qms_override=add_json_escape_parameters(parameters=qms_override).output,
       display_name=display_name,
       machine_type=machine_type,
       service_account=service_account,