Skip to content
Snippets Groups Projects
Verified Commit 91bfb128 authored by Yoann Schneider's avatar Yoann Schneider :tennis:
Browse files

retrieve IDs from config (revert me later please

parent 72623d76
No related branches found
No related tags found
1 merge request!2Implement worker
Pipeline #81805 passed
...@@ -76,15 +76,15 @@ class DatasetExtractor(BaseWorker): ...@@ -76,15 +76,15 @@ class DatasetExtractor(BaseWorker):
""" """
logger.info("Retrieving information from process_information") 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" assert train_folder_id, "A training folder id is necessary to use this worker"
self.training_folder_id = UUID(train_folder_id) 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" assert val_folder_id, "A validation folder id is necessary to use this worker"
self.validation_folder_id = UUID(val_folder_id) 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 self.testing_folder_id = UUID(test_folder_id) if test_folder_id else None
def initialize_database(self): def initialize_database(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment