Skip to content

Commit 753a2f1

Browse files
jsondaiGoogle Cloud Pipeline Components maintainers
authored andcommitted
fix(components): remove default prediction column names in evaluation regression component to fix issues with bigquery data source
Signed-off-by: Jason Dai <jsndai@google.com> PiperOrigin-RevId: 684488025
1 parent 391de8c commit 753a2f1

File tree

4 files changed

+4
-3
lines changed

4 files changed

+4
-3
lines changed

components/google-cloud/RELEASE.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
## Upcoming release
2+
* Remove default prediction column names in `v1.model_evaluation.regression_component` component to fix pipeline errors when using bigquery data source.
23

34
## Release 2.17.0
45
* Fix Gemini batch prediction support to `v1.model_evaluation.autosxs_pipeline` after output schema change.

components/google-cloud/google_cloud_pipeline_components/container/v1/aiplatform/remote_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def write_to_artifact(executor_input, text):
8888
# Add URI Prefix
8989
# "https://[location]-aiplatform.googleapis.com/API_VERSION/": For AI Platform resource names, current version is defined in AIPLATFORM_API_VERSION.
9090
if RESOURCE_NAME_PATTERN.match(text):
91-
location = re.findall(r'locations/([\w\-]+)', text)[0]
91+
location = re.findall('locations/([\w\-]+)', text)[0]
9292
uri_with_prefix = f'https://{location}-aiplatform.googleapis.com/{AIPLATFORM_API_VERSION}/{text}'
9393
metadata.update({'resourceName': text})
9494

components/google-cloud/google_cloud_pipeline_components/v1/hyperparameter_tuning_job/component.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def hyperparameter_tuning_job(
4242
project: str = _placeholders.PROJECT_ID_PLACEHOLDER,
4343
):
4444
# fmt: off
45-
r"""Creates a Vertex AI hyperparameter tuning job and waits for it to
45+
"""Creates a Vertex AI hyperparameter tuning job and waits for it to
4646
complete.
4747
4848
See [more information](https://cloud.google.com/vertex-ai/docs/training/using-hyperparameter-tuning).

components/google-cloud/google_cloud_pipeline_components/v1/model_evaluation/regression_component.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def model_evaluation_regression(
3636
ground_truth_format: str = 'jsonl',
3737
ground_truth_gcs_source: List[str] = [],
3838
ground_truth_bigquery_source: str = '',
39-
prediction_score_column: str = 'prediction.value',
39+
prediction_score_column: str = '',
4040
dataflow_service_account: str = '',
4141
dataflow_disk_size_gb: int = 50,
4242
dataflow_machine_type: str = 'n1-standard-4',

0 commit comments

Comments
 (0)