def _get_experiment_or_pipeline_resource_name( name: str, source: str, expected_schema: str ) -> str: """Get the full resource name of the Context representing an Experiment or Pipeline. Args: name (str): Name of the Experiment or Pipeline. source (str): Identify whether the this is an Experiment or a Pipeline. expected_schema (str): expected_schema identifies the expected schema used for Experiment or Pipeline. Returns: The full resource name of the Experiment or Pipeline Context. Raise: NotFound exception if experiment or pipeline does not exist. """ context = _Context(resource_name=name) if context.schema_title != expected_schema: raise ValueError( f"Please provide a valid {source} name. {name} is not a {source}." ) return context.resource_name
def test_init_context_with_id(self, get_context_mock): aiplatform.init(project=_TEST_PROJECT, location=_TEST_LOCATION) context._Context(resource_name=_TEST_CONTEXT_ID, metadata_store_id=_TEST_METADATA_STORE) get_context_mock.assert_called_once_with(name=_TEST_CONTEXT_NAME, retry=base._DEFAULT_RETRY)
def test_init_context(self, get_context_mock): aiplatform.init(project=_TEST_PROJECT) context._Context(resource_name=_TEST_CONTEXT_NAME) get_context_mock.assert_called_once_with(name=_TEST_CONTEXT_NAME, retry=base._DEFAULT_RETRY)