also retrieve test ID from process

worker_generic_training_dataset/worker.py

@@ -84,7 +84,7 @@ class DatasetExtractor(BaseWorker):
         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(test_folder_id) if test_folder_id else None
 
     def initialize_database(self):