fix(components): remove default prediction column names in evaluation…

… classification component to fix incorrect column names for bigquery data source PiperOrigin-RevId: 626116194
kubeflow · Apr 18, 2024 · 2d839fa · 2d839fa
1 parent 2a8d39e
commit 2d839fa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/components/google-cloud/RELEASE.md b/components/google-cloud/RELEASE.md
@@ -1,6 +1,7 @@
 ## Upcoming release
 * Use larger base reward model when tuning `text-bison@001`, `chat-bison@001` and `t5-xxl` with the `preview.llm.rlhf_pipeline`.
 * Move `preview.model_evaluation.autosxs_pipeline` to `v1.model_evaluation.autosxs_pipeline`.
+* Remove default prediction column names in `v1.model_evaluation.classification_component` component to fix pipeline errors when using bigquery data source.
 
 ## Release 2.13.1
 * Fix model name preprocess error, pass correct model to `ModelImportEvaluationOp` component in `v1.model_evaluation.evaluation_llm_text_generation_pipeline` and `v1.model_evaluation.evaluation_llm_classification_pipeline`.

diff --git a/...le-cloud/google_cloud_pipeline_components/v1/model_evaluation/classification_component.py b/...le-cloud/google_cloud_pipeline_components/v1/model_evaluation/classification_component.py
@@ -38,8 +38,8 @@ def model_evaluation_classification(
     ground_truth_bigquery_source: str = '',
     classification_type: str = 'multiclass',
     class_labels: List[str] = [],
-    prediction_score_column: str = 'prediction.scores',
-    prediction_label_column: str = 'prediction.classes',
+    prediction_score_column: str = '',
+    prediction_label_column: str = '',
     slicing_specs: List[Any] = [],
     positive_classes: List[str] = [],
     dataflow_service_account: str = '',