From b2336145de5f6bedc910d39697faffd5fd34abd0 Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Wed, 19 Sep 2018 15:10:35 +0200
Subject: [PATCH] catch KeyError as well as IOError in _load_data
---
toolkit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/toolkit.py b/toolkit.py
index b224fed..be90cc6 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -409,7 +409,7 @@ class ClassificationProject(object):
# if those don't exist, we need to load them from ROOT trees first
self._load_from_hdf5(*self.dataset_names_tree)
- except KeyError:
+ except (KeyError, IOError):
logger.info("Couldn't load all datasets - reading from ROOT trees")
--
GitLab