fixed possible DivByZero bug in benchmarking

alrevuelta · Mar 9, 2021 · 8b6f9ed · 8b6f9ed
1 parent d464b9e
commit 8b6f9ed
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/benchmarking.py b/tests/benchmarking.py
@@ -25,6 +25,7 @@ def benchmark_model(id, path, io, n_inputs, n_outputs, n_runs=1):
         elif result == 0:
             print("[Warning] Measured time is 0. Perhaps trying to measure a very small time."+
                   "There is a known issue with time library in Windows.")
+            model_times.append(result)
         else:
             raise Exception("The output of the model doesn't match the expected")