From 05c3559e2dfd015b2e7f2deca7aa9f1692c84ecb Mon Sep 17 00:00:00 2001 From: Florian Fischer Date: Tue, 24 Jul 2018 10:28:20 +0200 Subject: fix save and load --- benchmark.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'benchmark.py') diff --git a/benchmark.py b/benchmark.py index 4b565e3..c68dce7 100644 --- a/benchmark.py +++ b/benchmark.py @@ -2,16 +2,16 @@ import pickle class Benchmark (object): def save(self, path=None, verbose=False): - f = path if path else self.file_name + ".save" + f = path if path else self.name + ".save" if verbose: - print("Saving results to:", self.file_name + ".save") + print("Saving results to:", self.name + ".save") with open(f, "wb") as f: pickle.dump(self.results, f) def load(self, path=None, verbose=False): - f = path if path else self.file_name + ".save" + f = path if path else self.name + ".save" if verbose: - print("Loading results from:", self.file_name + ".save") + print("Loading results from:", self.name + ".save") with open(f, "rb") as f: self.results = pickle.load(f) -- cgit v1.2.3