2626from .profilers .cpu_profiler import CPUProfiler
2727from .profilers .allocation_profiler import AllocationProfiler
2828from .profilers .block_profiler import BlockProfiler
29- from .profilers .tf_profiler import TFProfiler
3029
3130
3231class Span (object ):
3332
3433 def __init__ (self , stop_func = None ):
3534 if stop_func :
3635 self .stop_func = stop_func
36+ else :
37+ self .stop_func = None
3738
3839
3940 def stop (self ):
@@ -51,7 +52,7 @@ def __exit__(self, exc_type, exc_value, traceback):
5152
5253class Agent (object ):
5354
54- AGENT_VERSION = "1.2.4 "
55+ AGENT_VERSION = "1.2.6 "
5556 SAAS_DASHBOARD_ADDRESS = "https://agent-api.stackimpact.com"
5657
5758 def __init__ (self , ** kwargs ):
@@ -100,15 +101,6 @@ def __init__(self, **kwargs):
100101 config .report_interval = 120
101102 self .block_reporter = ProfileReporter (self , BlockProfiler (self ), config )
102103
103- config = ProfilerConfig ()
104- config .log_prefix = 'TensorFlow profiler'
105- config .max_profile_duration = 20
106- config .max_span_duration = 5
107- config .max_span_count = 30
108- config .span_interval = 20
109- config .report_interval = 120
110- self .tf_reporter = ProfileReporter (self , TFProfiler (self ), config )
111-
112104 self .options = None
113105
114106
@@ -160,7 +152,6 @@ def start(self, **kwargs):
160152 self .cpu_reporter .setup ()
161153 self .allocation_reporter .setup ()
162154 self .block_reporter .setup ()
163- self .tf_reporter .setup ()
164155 self .span_reporter .setup ()
165156 self .error_reporter .setup ()
166157 self .process_reporter .setup ()
@@ -211,7 +202,6 @@ def enable(self):
211202 self .cpu_reporter .start ()
212203 self .allocation_reporter .start ()
213204 self .block_reporter .start ()
214- self .tf_reporter .start ()
215205 self .span_reporter .start ()
216206 self .error_reporter .start ()
217207 self .process_reporter .start ()
@@ -223,7 +213,6 @@ def disable(self):
223213 self .cpu_reporter .stop ()
224214 self .allocation_reporter .stop ()
225215 self .block_reporter .stop ()
226- self .tf_reporter .stop ()
227216 self .span_reporter .stop ()
228217 self .error_reporter .stop ()
229218 self .process_reporter .stop ()
@@ -244,8 +233,6 @@ def profile(self, name='Default'):
244233 active_reporters .append (self .allocation_reporter )
245234 if self .block_reporter .started :
246235 active_reporters .append (self .block_reporter )
247- if self .tf_reporter .started :
248- active_reporters .append (self .tf_reporter )
249236
250237 if len (active_reporters ) > 0 :
251238 selected_reporter = active_reporters [int (math .floor (random .random () * len (active_reporters )))]
@@ -318,14 +305,6 @@ def stop_block_profiler(self):
318305 self ._stop_profiler (self .block_reporter )
319306
320307
321- def start_tf_profiler (self ):
322- self ._start_profiler (self .tf_reporter )
323-
324-
325- def stop_tf_profiler (self ):
326- self ._stop_profiler (self .tf_reporter )
327-
328-
329308 def destroy (self ):
330309 if not self .agent_started :
331310 self .log ('Agent has not been started' )
@@ -340,15 +319,13 @@ def destroy(self):
340319 self .cpu_reporter .stop ()
341320 self .allocation_reporter .stop ()
342321 self .block_reporter .stop ()
343- self .tf_reporter .stop ()
344322 self .error_reporter .stop ()
345323 self .span_reporter .stop ()
346324 self .process_reporter .stop ()
347325
348326 self .cpu_reporter .destroy ()
349327 self .allocation_reporter .destroy ()
350328 self .block_reporter .destroy ()
351- self .tf_reporter .destroy ()
352329 self .error_reporter .destroy ()
353330 self .span_reporter .destroy ()
354331 self .process_reporter .destroy ()
0 commit comments