summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--buildstream/_artifactcache.py2
-rw-r--r--buildstream/element.py3
2 files changed, 4 insertions, 1 deletions
diff --git a/buildstream/_artifactcache.py b/buildstream/_artifactcache.py
index 5df380457..4d074c29d 100644
--- a/buildstream/_artifactcache.py
+++ b/buildstream/_artifactcache.py
@@ -542,7 +542,7 @@ class ArtifactCache(BaseCache):
keys = list(keys)
if not keys:
keys = [element._get_cache_key()]
- artifacts = list(map(self.get_artifact_proto, list(map(element.get_artifact_name, keys))))
+ artifacts = list(map(element.get_artifact_proto, keys))
# check the artifacts are the same for each key
# unsure how necessary this is
artifact = artifacts[0]
diff --git a/buildstream/element.py b/buildstream/element.py
index ba8e93d88..0f4560642 100644
--- a/buildstream/element.py
+++ b/buildstream/element.py
@@ -455,6 +455,9 @@ class Element(Plugin):
yield from visit(self, scope, visited)
+ def get_artifact_proto(self, key):
+ return self.__artifact._get_proto()
+
def search(self, scope, name):
"""Search for a dependency by name