Skip to content

Commit

Permalink
feat(component): internal
Browse files Browse the repository at this point in the history
Signed-off-by: Googler <[email protected]>
PiperOrigin-RevId: 663872006
  • Loading branch information
Googler committed Aug 16, 2024
1 parent 8041ea8 commit 383cbcd
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 17 deletions.
1 change: 1 addition & 0 deletions components/google-cloud/RELEASE.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
## Upcoming release
* Fix to model batch explanation component for Structured Data pipelines; image bump.
* Add dynamic support for boot_disk_type, boot_disk_size in `preview.custom_job.utils.create_custom_training_job_from_component`.
* Remove preflight validations temporarily.

## Release 2.16.0
* Updated the Starry Net pipeline's template gallery description, and added dataprep_nan_threshold and dataprep_zero_threshold args to the Starry Net pipeline.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -15,12 +15,10 @@

from typing import Dict, List, NamedTuple

from google_cloud_pipeline_components import google_template_metadata
from google_cloud_pipeline_components._implementation.model_evaluation import LLMEvaluationPreprocessorOp
from google_cloud_pipeline_components._implementation.model_evaluation import LLMEvaluationTextGenerationOp
from google_cloud_pipeline_components._implementation.model_evaluation import ModelNamePreprocessorOp
from google_cloud_pipeline_components.preview.model_evaluation.model_evaluation_import_component import model_evaluation_import as ModelImportEvaluationOp
from google_cloud_pipeline_components.proto import template_metadata_pb2
from google_cloud_pipeline_components.types.artifact_types import VertexModel
from google_cloud_pipeline_components.v1.batch_predict_job import ModelBatchPredictOp
from kfp import dsl
Expand All @@ -31,21 +29,6 @@

_PIPELINE_NAME = 'evaluation-llm-text-generation-pipeline'

output_gcs_validation = template_metadata_pb2.GoogleCloudStorageValidation(
gcs_uri='{{$.parameter.batch_predict_gcs_destination_output_uri}}',
is_input=False,
default_service_account='{{$.pipeline_google_cloud_project_number}}[email protected]',
override_placeholder='{{$.parameter.service_account}}',
)


@google_template_metadata.set_template_metadata(
template_metadata=template_metadata_pb2.TemplateMetadata(
preflight_validations=template_metadata_pb2.ValidationItems(
gcs_validations=[output_gcs_validation]
)
)
)
@dsl.pipeline(name=_PIPELINE_NAME)
def evaluation_llm_text_generation_pipeline( # pylint: disable=dangerous-default-value
project: str,
Expand Down

0 comments on commit 383cbcd

Please sign in to comment.