def query_input_and_output_artifacts(self) -> Sequence[artifact._Artifact]: """query the input and output artifacts connected to the execution. Returns: A Sequence of _Artifacts """ try: artifacts = self.api_client.query_execution_inputs_and_outputs( execution=self.resource_name).artifacts except exceptions.NotFound: return [] return [ artifact._Artifact( resource=metadata_artifact, project=self.project, location=self.location, credentials=self.credentials, ) for metadata_artifact in artifacts ]
def test_init_artifact_with_id(self, get_artifact_mock): aiplatform.init(project=_TEST_PROJECT, location=_TEST_LOCATION) artifact._Artifact(resource_name=_TEST_ARTIFACT_ID, metadata_store_id=_TEST_METADATA_STORE) get_artifact_mock.assert_called_once_with(name=_TEST_ARTIFACT_NAME, retry=base._DEFAULT_RETRY)
def test_init_artifact(self, get_artifact_mock): aiplatform.init(project=_TEST_PROJECT) artifact._Artifact(resource_name=_TEST_ARTIFACT_NAME) get_artifact_mock.assert_called_once_with(name=_TEST_ARTIFACT_NAME, retry=base._DEFAULT_RETRY)