Skip to content
Snippets Groups Projects
Verified Commit f904431e authored by Erwan Rouchet's avatar Erwan Rouchet
Browse files

Restore get_worker_version_slug

parent 99194803
No related branches found
No related tags found
1 merge request!72Drop DataSource handling
Pipeline #78395 passed
......@@ -14,6 +14,13 @@ class WorkerVersionMixin(object):
return worker_version
def get_worker_version_slug(self, worker_version_id: str) -> str:
"""
Get worker version slug from cache if possible, otherwise make API request
"""
worker_version = self.get_worker_version(worker_version_id)
return worker_version["worker"]["slug"]
def get_ml_result_slug(self, ml_result) -> str:
"""
Helper function to get the worker slug from element, classification or transcription.
......@@ -36,5 +43,4 @@ class WorkerVersionMixin(object):
if worker_version_id is None:
return
worker_version = self.get_worker_version(worker_version_id)
return worker_version["worker"]["slug"]
return self.get_worker_version_slug(worker_version_id)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment