Skip to content
Snippets Groups Projects

Drop DataSource handling

Merged Erwan Rouchet requested to merge drop-datasource into master
All threads resolved!
2 files
+ 6
9
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -14,7 +14,6 @@ class WorkerVersionMixin(object):
return worker_version
def get_ml_result_slug(self, ml_result) -> str:
"""
Helper function to get the worker slug from element, classification or transcription.
@@ -24,13 +23,13 @@ class WorkerVersionMixin(object):
"""
# Handle cached models
if hasattr(ml_result, 'worker_version_id'):
if hasattr(ml_result, "worker_version_id"):
worker_version_id = ml_result.worker_version_id
elif "worker_version" in ml_result:
worker_version_id = ml_result['worker_version']
worker_version_id = ml_result["worker_version"]
# transcriptions have worker_version_id but elements have worker_version
elif "worker_version_id" in ml_result:
worker_version_id = ml_result['worker_version_id']
worker_version_id = ml_result["worker_version_id"]
else:
raise ValueError(f"Unable to get slug from: {ml_result}")
@@ -38,4 +37,4 @@ class WorkerVersionMixin(object):
return
worker_version = self.get_worker_version(worker_version_id)
return worker_version['worker']['slug']
return worker_version["worker"]["slug"]
Loading