Refactor log_config assignment to include both 'log' and 'metafeatures' in GenerateEventLogs class

gedi/generator.py

@@ -170,8 +170,12 @@ class GenerateEventLogs():
                 print(f"INFO: Generator starting at {start.strftime('%H:%M:%S')} using {num_cores} cores for {len(tasks)} tasks...")
                 random.seed(RANDOM_SEED)
                 log_config = p.map(self.generator_wrapper, [(index, row) for index, row in tasks.iterrows()])
-            self.log_config = log_config
-            # self.log_config = [config['metafeatures'] for config in log_config if 'metafeatures' in config]
+            # self.log_config = log_config
+            self.log_config = [
+                        {'log': config.get('log'), 'metafeatures': config.get('metafeatures')}
+                        for config in log_config
+                        if 'metafeatures' in config and 'log' in config
+                    ]
 
         else:
             random.seed(RANDOM_SEED)