def simple_timeit()

in dags/multipod/legacy_tests/gpt1-like.py [0:0]

• 12 lines of code
• 3 McCabe index (conditional complexity)

def simple_timeit(f, tries=10, verbose=True):
  outcomes = []
  f()  # warm it up!
  for i in range(tries):
    s = datetime.datetime.now()
    r = f()
    e = datetime.datetime.now()
    outcomes.append((e - s).total_seconds())
  average_time = sum(outcomes) / len(outcomes)
  if verbose:
    print(f"average time: {average_time}, timings (seconds) {outcomes}")
  return average_time