Aryasomayajula, Sai Anirudh [External] commited on
Commit
a607c6a
·
1 Parent(s): e27ec77

Add memory profiling and clear method to GenerateEventLogs class

Browse files
Files changed (1) hide show
  1. gedi/generator.py +10 -0
gedi/generator.py CHANGED
@@ -25,6 +25,7 @@ from gedi.utils.param_keys.generator import GENERATOR_PARAMS, EXPERIMENT, CONFIG
25
  import xml.etree.ElementTree as ET
26
  import re
27
  from xml.dom import minidom
 
28
 
29
  """
30
  Parameters
@@ -188,6 +189,15 @@ class GenerateEventLogs():
188
  print(f" Saved generated logs in {self.output_path}")
189
  print("========================= ~ Generator ==========================")
190
 
 
 
 
 
 
 
 
 
 
191
  def generator_wrapper(self, task):
192
  try:
193
  identifier = [x for x in task[1] if isinstance(x, str)][0]
 
25
  import xml.etree.ElementTree as ET
26
  import re
27
  from xml.dom import minidom
28
+ from memory_profiler import profile
29
 
30
  """
31
  Parameters
 
189
  print(f" Saved generated logs in {self.output_path}")
190
  print("========================= ~ Generator ==========================")
191
 
192
+
193
+ def clear(self):
194
+ self.log_config = None
195
+ self.configs = None
196
+ self.params = None
197
+ self.output_path = None
198
+ self.feature_keys = None
199
+
200
+
201
  def generator_wrapper(self, task):
202
  try:
203
  identifier = [x for x in task[1] if isinstance(x, str)][0]