From 76dc26398cb9cce20b9ba099b46350d4c587c01a Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Thu, 11 May 2023 18:25:25 +0200
Subject: [PATCH] remove debug modifs
---
ci/build.sh | 1 -
worker_generic_training_dataset/worker.py | 6 +++---
2 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/ci/build.sh b/ci/build.sh
index 7c43de4..f29f50f 100755
--- a/ci/build.sh
+++ b/ci/build.sh
@@ -5,7 +5,6 @@
# Will automatically login to a registry if CI_REGISTRY, CI_REGISTRY_USER and CI_REGISTRY_PASSWORD are set.
# Will only push an image if $CI_REGISTRY is set.
-VERSION="POC"
if [ -z "$VERSION" ]; then
VERSION=${CI_COMMIT_TAG:-latest}
fi
diff --git a/worker_generic_training_dataset/worker.py b/worker_generic_training_dataset/worker.py
index 7c20093..f547e42 100644
--- a/worker_generic_training_dataset/worker.py
+++ b/worker_generic_training_dataset/worker.py
@@ -77,15 +77,15 @@ class DatasetExtractor(BaseWorker):
"""
logger.info("Retrieving information from process_information")
- train_folder_id = self.config.get("train_folder_id")
+ train_folder_id = self.process_information.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.config.get("validation_folder_id")
+ val_folder_id = self.process_information.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.config.get("test_folder_id")
+ test_folder_id = self.process_information.get("test_folder_id")
self.testing_folder_id: UUID | None = (
UUID(test_folder_id) if test_folder_id else None
)
--
GitLab