load and save from/to specific file

1 files changed, 6 insertions, 4 deletions

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)