From e29de27e578432762ba4afd225d0331a8c66b7f7 Mon Sep 17 00:00:00 2001 From: Florian Fischer Date: Mon, 23 Jul 2018 11:26:50 +0200 Subject: load and save from/to specific file --- benchmark.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'benchmark.py') diff --git a/benchmark.py b/benchmark.py index 9b7316a..4b565e3 100644 --- a/benchmark.py +++ b/benchmark.py @@ -1,15 +1,17 @@ import pickle class Benchmark (object): - def save(self, verbose=False): + def save(self, path=None, verbose=False): + f = path if path else self.file_name + ".save" if verbose: print("Saving results to:", self.file_name + ".save") - with open(self.file_name + ".save", "wb") as f: + with open(f, "wb") as f: pickle.dump(self.results, f) - def load(self, verbose=False): + def load(self, path=None, verbose=False): + f = path if path else self.file_name + ".save" if verbose: print("Loading results from:", self.file_name + ".save") - with open(self.file_name + ".save", "rb") as f: + with open(f, "rb") as f: self.results = pickle.load(f) -- cgit v1.2.3