import numpy as np import time from collections import deque # timer class Timer: def __init__(self, interval=10): self.value = 0.0 self.start = time.time() self.buffer = deque(maxlen=interval) def __enter__(self): self.start = time.time() def __exit__(self, exc_type, exc_val, exc_tb): self.value = time.time() - self.start self.buffer.append(self.value) @property def now(self): """Return the seconds elapsed since initializing this class""" return time.time() - self.start @property def avg(self): return np.mean(self.buffer, dtype=float)