Spaces:
Running
Running
Aryasomayajula, Sai Anirudh [External]
commited on
Commit
·
f303bf3
1
Parent(s):
2d5450f
Refactor log_config assignment to filter for 'metafeatures' and enhance clear method with a log message
Browse files- gedi/generator.py +3 -2
gedi/generator.py
CHANGED
@@ -170,7 +170,7 @@ class GenerateEventLogs():
|
|
170 |
print(f"INFO: Generator starting at {start.strftime('%H:%M:%S')} using {num_cores} cores for {len(tasks)} tasks...")
|
171 |
random.seed(RANDOM_SEED)
|
172 |
log_config = p.map(self.generator_wrapper, [(index, row) for index, row in tasks.iterrows()])
|
173 |
-
self.log_config = log_config
|
174 |
|
175 |
else:
|
176 |
random.seed(RANDOM_SEED)
|
@@ -178,7 +178,7 @@ class GenerateEventLogs():
|
|
178 |
if type(self.configs) is not list:
|
179 |
self.configs = [self.configs]
|
180 |
temp = self.generate_optimized_log(self.configs[0])
|
181 |
-
self.log_config = [temp]
|
182 |
save_path = get_output_key_value_location(self.params[EXPERIMENT],
|
183 |
self.output_path, "genEL")+".xes"
|
184 |
write_xes(temp['log'], save_path)
|
@@ -190,6 +190,7 @@ class GenerateEventLogs():
|
|
190 |
|
191 |
|
192 |
def clear(self):
|
|
|
193 |
self.log_config = None
|
194 |
self.configs = None
|
195 |
self.params = None
|
|
|
170 |
print(f"INFO: Generator starting at {start.strftime('%H:%M:%S')} using {num_cores} cores for {len(tasks)} tasks...")
|
171 |
random.seed(RANDOM_SEED)
|
172 |
log_config = p.map(self.generator_wrapper, [(index, row) for index, row in tasks.iterrows()])
|
173 |
+
self.log_config = [config['metafeatures'] for config in log_config if 'metafeatures' in config]
|
174 |
|
175 |
else:
|
176 |
random.seed(RANDOM_SEED)
|
|
|
178 |
if type(self.configs) is not list:
|
179 |
self.configs = [self.configs]
|
180 |
temp = self.generate_optimized_log(self.configs[0])
|
181 |
+
self.log_config = [temp['metafeatures']] if 'metafeatures' in temp else []
|
182 |
save_path = get_output_key_value_location(self.params[EXPERIMENT],
|
183 |
self.output_path, "genEL")+".xes"
|
184 |
write_xes(temp['log'], save_path)
|
|
|
190 |
|
191 |
|
192 |
def clear(self):
|
193 |
+
print("Clearing parameters...")
|
194 |
self.log_config = None
|
195 |
self.configs = None
|
196 |
self.params = None
|