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
Files changed (1) hide show
  1. 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