diff options
Diffstat (limited to 'src/benchmark.py')
| -rw-r--r-- | src/benchmark.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/benchmark.py b/src/benchmark.py index 0f5dbdb..a0da6a0 100644 --- a/src/benchmark.py +++ b/src/benchmark.py @@ -62,7 +62,13 @@ class Benchmark (object): pickle.dump(save_data, f) def load(self, path=None, verbose=False): - f = path if path else self.name + ".save" + if not path: + f = self.name + ".save" + else: + if os.path.isdir(path): + f = os.path.join(path, self.name + ".save") + else: + f = path if verbose: print("Loading results from:", self.name + ".save") with open(f, "rb") as f: |