Spaces:
Running
Running
Aryasomayajula, Sai Anirudh [External]
commited on
Commit
·
a607c6a
1
Parent(s):
e27ec77
Add memory profiling and clear method to GenerateEventLogs class
Browse files- 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]
|