diff --git a/benchmark.py b/benchmark.py index 4b565e3ca3458773e766f4bd6b24919a23d3968b..c68dce754cb59312d5ffd7cf8320b030f6d36b47 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)