From 3d676dfc064189f906c8afe70f90dde9494baada Mon Sep 17 00:00:00 2001 From: Florian Fischer Date: Tue, 22 Jan 2019 15:09:57 +0100 Subject: add load path option and don't fail if result directories exists already --- src/benchmark.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'src') 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: -- cgit v1.2.3