From e326c11a541d5e74937f1f885890b0b1b6cd88be Mon Sep 17 00:00:00 2001
From: Erwan Rouchet <rouchet@teklia.com>
Date: Mon, 17 Feb 2025 17:37:10 +0100
Subject: [PATCH] Declare a system worker for generic dataset extraction

---
 arkindex/process/models.py  | 1 +
 arkindex/system_workers.yml | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/arkindex/process/models.py b/arkindex/process/models.py
index 920823bb6..f9fa1efaa 100644
--- a/arkindex/process/models.py
+++ b/arkindex/process/models.py
@@ -723,6 +723,7 @@ class ArkindexFeature(Enum):
     InitElements = "init_elements"
     FileImport = "file_import"
     S3Ingest = "s3_ingest"
+    DatasetExtractor = "dataset_extractor"
     # When adding a new export worker, don't forget to also update the ExportFormat enum and the
     # FEATURE_FORMAT_MAP dictionary which maps export formats to arkindex features
     ExportPDF = "pdf_export"
diff --git a/arkindex/system_workers.yml b/arkindex/system_workers.yml
index 8e87093ba..e8f59083a 100644
--- a/arkindex/system_workers.yml
+++ b/arkindex/system_workers.yml
@@ -31,3 +31,8 @@ features:
             name: arkindex/workers/export
             version: 0.2.1
             slug: csv-export
+    dataset_extractor:
+        teklia_worker:
+            name: arkindex/workers/generic-training-dataset
+            version: 0.3.0
+            slug: generic-training-dataset
-- 
GitLab