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

also retrieve test ID from process

parent 6a4d1900
No related branches found
No related tags found
1 merge request!2Implement worker
Hide whitespace changes
Inline Side-by-side
worker_generic_training_dataset/worker.py
+ 1
− 1
View file @ 72623d76
...@@ -84,7 +84,7 @@ class DatasetExtractor(BaseWorker): ...@@ -84,7 +84,7 @@ class DatasetExtractor(BaseWorker):
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.config.get("test_folder_id") test_folder_id = self.process_information.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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment