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/'