From 1782ee2d899f7e33fa997a740a3678bf7555fbb6 Mon Sep 17 00:00:00 2001 From: Jason Dai Date: Thu, 25 May 2023 16:44:16 -0700 Subject: [PATCH] feat(components): add a check for model garden models and skip eval import PiperOrigin-RevId: 535431767 --- .../experimental/evaluation/import_model_evaluation.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/components/google-cloud/google_cloud_pipeline_components/container/experimental/evaluation/import_model_evaluation.py b/components/google-cloud/google_cloud_pipeline_components/container/experimental/evaluation/import_model_evaluation.py index a473464bb4..1bcf928eb6 100644 --- a/components/google-cloud/google_cloud_pipeline_components/container/experimental/evaluation/import_model_evaluation.py +++ b/components/google-cloud/google_cloud_pipeline_components/container/experimental/evaluation/import_model_evaluation.py @@ -18,7 +18,7 @@ import json import logging import os import sys -from typing import Any, Dict, List, Optional, Tuple, Iterable +from typing import Any, Dict, Iterable, List, Optional, Tuple from google.api_core import gapic_v1 from google.cloud import aiplatform @@ -135,6 +135,9 @@ def main(argv): """Calls ModelService.ImportModelEvaluation.""" parsed_args, _ = parser.parse_known_args(argv) + if parsed_args.model_name.startswith('publishers'): + return + _, project_id, _, location, _, model_id = parsed_args.model_name.split('/') api_endpoint = location + '-aiplatform.googleapis.com' resource_uri_prefix = f'https://{api_endpoint}/v1/'