From 91bfb12845a3dab39f45ca2638829e6325f53018 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Fri, 5 May 2023 13:52:49 +0200
Subject: [PATCH] retrieve IDs from config (revert me later please
---
worker_generic_training_dataset/worker.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/worker_generic_training_dataset/worker.py b/worker_generic_training_dataset/worker.py
index 7270050..30120a1 100644
--- a/worker_generic_training_dataset/worker.py
+++ b/worker_generic_training_dataset/worker.py
@@ -76,15 +76,15 @@ class DatasetExtractor(BaseWorker):
"""
logger.info("Retrieving information from process_information")
- train_folder_id = self.process_information.get("train_folder_id")
+ train_folder_id = self.config.get("train_folder_id")
assert train_folder_id, "A training folder id is necessary to use this worker"
self.training_folder_id = UUID(train_folder_id)
- val_folder_id = self.process_information.get("validation_folder_id")
+ val_folder_id = self.config.get("validation_folder_id")
assert val_folder_id, "A validation folder id is necessary to use this worker"
self.validation_folder_id = UUID(val_folder_id)
- test_folder_id = self.process_information.get("test_folder_id")
+ test_folder_id = self.config.get("test_folder_id")
self.testing_folder_id = UUID(test_folder_id) if test_folder_id else None
def initialize_database(self):
--
GitLab