Skip to content

Commit ec7d462

Browse files
author
Benjamin
committed
additional cleanup
1 parent 0198978 commit ec7d462

File tree

1 file changed

+4
-20
lines changed

1 file changed

+4
-20
lines changed

Diff for: src/deepsparse/utils/timer.py

+4-20
Original file line numberDiff line numberDiff line change
@@ -170,28 +170,12 @@ class TimerManager:
170170
"""
171171

172172
def __init__(self, enabled: bool = True, multi: bool = False):
173-
self._multi = multi
174-
self._enabled = enabled
173+
self.multi = multi
174+
self.enabled = enabled
175175
self._timers = []
176176

177177
def __repr__(self):
178-
return f"PipelineTimer({self.times})"
179-
180-
@property
181-
def enabled(self):
182-
return self._enabled
183-
184-
@enabled.setter
185-
def enabled(self, value):
186-
self._enabled = value
187-
188-
@property
189-
def multi(self):
190-
return self._multi
191-
192-
@multi.setter
193-
def multi(self, value):
194-
self._multi = value
178+
return f"TimerManager({self.times})"
195179

196180
@property
197181
def timers(self) -> List[StagedTimer]:
@@ -226,7 +210,7 @@ def all_times(self) -> Dict[str, List[float]]:
226210
return all_times
227211

228212
def new_timer(self) -> StagedTimer:
229-
timer = StagedTimer()
213+
timer = StagedTimer(enabled=self.enabled)
230214

231215
if self.multi:
232216
self._timers.append(timer)

0 commit comments

Comments
 (0)