Skip to content

Commit 449c304

Browse files
TheMichaelHuGoogle Cloud Pipeline Components maintainers
authored andcommitted
fix(components): Use PipelineJob location in AutoSxS components, add init file
PiperOrigin-RevId: 607055407
1 parent ae62b11 commit 449c304

File tree

3 files changed

+15
-4
lines changed

3 files changed

+15
-4
lines changed
Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
# Copyright 2024 The Kubeflow Authors. All Rights Reserved.
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.

components/google-cloud/google_cloud_pipeline_components/_implementation/llm/model_evaluation_text_generation_pairwise.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,7 @@ def model_evaluation_text_generation_pairwise(
4848
"""
4949
return gcpc_utils.build_serverless_customjob_container_spec(
5050
project=_placeholders.PROJECT_ID_PLACEHOLDER,
51-
# Hardcode location to us-central1 for text-bison availability.
52-
location='us-central1',
51+
location=_placeholders.LOCATION_PLACEHOLDER,
5352
custom_job_payload=utils.build_payload(
5453
display_name='model_evaluation_text_generation_pairwise',
5554
machine_type='n1-standard-4',

components/google-cloud/google_cloud_pipeline_components/_implementation/llm/online_evaluation_pairwise.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,8 +74,7 @@ def online_evaluation_pairwise(
7474
"""
7575
return gcpc_utils.build_serverless_customjob_container_spec(
7676
project=_placeholders.PROJECT_ID_PLACEHOLDER,
77-
# Hardcode location to us-central1 for text-bison availability.
78-
location='us-central1',
77+
location=_placeholders.LOCATION_PLACEHOLDER,
7978
custom_job_payload=utils.build_payload(
8079
display_name='online_evaluation_pairwise',
8180
machine_type='n1-standard-4',

0 commit comments

Comments
 (0)