diff options
Diffstat (limited to 'benchmark.py')
| -rw-r--r-- | benchmark.py | 8 |
1 files changed, 4 insertions, 4 deletions
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) |