From 74878db39a9247936d78def423c74c95ea088d00 Mon Sep 17 00:00:00 2001 From: Florian Fischer Date: Sun, 28 Apr 2019 17:01:44 +0200 Subject: don't fail if NaN is in eval --- src/benchmark.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/benchmark.py b/src/benchmark.py index 971c4a8..4bab146 100644 --- a/src/benchmark.py +++ b/src/benchmark.py @@ -13,6 +13,10 @@ import src.globalvars from src.util import * +# This is useful when evaluating strings in the plot functionsi. str(np.NaN) == "nan" +nan = np.NaN + + class Benchmark (object): perf_allowed = None @@ -458,7 +462,9 @@ class Benchmark (object): norm_mean = eval(yval.format(**self.results["stats"][scale][perm]["mean"])) y_vals.append(mean / norm_mean) else: - y_vals.append(eval(yval.format(**self.results["stats"][allocator][perm]["mean"]))) + eval_dict = self.results["stats"][allocator][perm]["mean"] + eval_str = yval.format(**eval_dict) + y_vals.append(eval(eval_str)) plt.plot(x_vals, y_vals, marker='.', linestyle='-', -- cgit v1.2.3