fix(components): Use PipelineJob location in AutoSxS components, add …

…init file PiperOrigin-RevId: 606661626
kubeflow · Feb 14, 2024 · cfbb7bb · cfbb7bb
1 parent 3f0fc06
commit cfbb7bb
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 4 deletions.
diff --git a/...s/google-cloud/google_cloud_pipeline_components/_implementation/llm/generated/__init__.py b/...s/google-cloud/google_cloud_pipeline_components/_implementation/llm/generated/__init__.py
@@ -0,0 +1,13 @@
+# Copyright 2024 The Kubeflow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
diff --git a/...loud_pipeline_components/_implementation/llm/model_evaluation_text_generation_pairwise.py b/...loud_pipeline_components/_implementation/llm/model_evaluation_text_generation_pairwise.py
@@ -48,8 +48,7 @@ def model_evaluation_text_generation_pairwise(
   """
   return gcpc_utils.build_serverless_customjob_container_spec(
       project=_placeholders.PROJECT_ID_PLACEHOLDER,
-      # Hardcode location to us-central1 for text-bison availability.
-      location='us-central1',
+      location=_placeholders.LOCATION_PLACEHOLDER,
       custom_job_payload=utils.build_payload(
           display_name='model_evaluation_text_generation_pairwise',
           machine_type='n1-standard-4',

diff --git a/...-cloud/google_cloud_pipeline_components/_implementation/llm/online_evaluation_pairwise.py b/...-cloud/google_cloud_pipeline_components/_implementation/llm/online_evaluation_pairwise.py
@@ -74,8 +74,7 @@ def online_evaluation_pairwise(
   """
   return gcpc_utils.build_serverless_customjob_container_spec(
       project=_placeholders.PROJECT_ID_PLACEHOLDER,
-      # Hardcode location to us-central1 for text-bison availability.
-      location='us-central1',
+      location=_placeholders.LOCATION_PLACEHOLDER,
       custom_job_payload=utils.build_payload(
           display_name='online_evaluation_pairwise',
           machine_type='n1-standard-4',