def _get_artifacts_with_type_and_pipeline(
self, type_name: Text,
pipeline_name: Text) -> List[metadata_store_pb2.Artifact]:
"""Helper function returns artifacts of specified pipeline and type."""
request = metadata_store_service_pb2.GetArtifactsByTypeRequest(
type_name=type_name)
all_artifacts = self._stub.GetArtifactsByType(request).artifacts
return [
artifact for artifact in all_artifacts
if artifact.custom_properties['pipeline_name'].string_value ==
pipeline_name
]
def get_artifacts_by_type(
self, type_name: Text) -> List[metadata_store_pb2.Artifact]:
"""Gets all the artifacts of a given type."""
request = metadata_store_service_pb2.GetArtifactsByTypeRequest()
request.type_name = type_name
response = metadata_store_service_pb2.GetArtifactsByTypeResponse()
self._call('GetArtifactsByType', request, response)
result = []
for x in response.artifacts:
result.append(x)
return result
def _get_artifacts_with_type(
self, type_name: Text) -> List[metadata_store_pb2.Artifact]:
"""Helper function returns artifacts with given type."""
request = metadata_store_service_pb2.GetArtifactsByTypeRequest(
type_name=type_name)
return self._stub.GetArtifactsByType(request).artifacts
