Skip to content
Snippets Groups Projects

Draft: Refactor and implement API version of the worker

Open Yoann Schneider requested to merge new-api-worker into main
2 files
+ 14
4
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -156,14 +156,14 @@ class Extractor(DatasetWorker):
"""
logger.info(f"Processing element ({element})")
polygon = element.polygon
if isinstance(element, Element):
image = element.image
polygon = element.polygon
wk_version = get_object_id(element.worker_version)
wk_run = get_object_id(element.worker_run)
else:
image = element.zone.image
polygon = element.zone.polygon
wk_version = (
element.worker_version
if hasattr(element, "worker_version")
@@ -229,7 +229,7 @@ class Extractor(DatasetWorker):
f"Linking element {cached_element.id} to dataset ({self.cached_dataset.id})"
)
with cache_database.atomic():
cached_element: CachedDatasetElement = CachedDatasetElement.create(
CachedDatasetElement.create(
id=uuid.uuid4(),
element=cached_element,
dataset=self.cached_dataset,
Loading