Save reference to returned data reading class on load

SamStudio8 · Aug 12, 2014 · d2c0367 · d2c0367
1 parent 5b8c80e
commit d2c0367
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/frontier/frontier.py b/frontier/frontier.py
@@ -81,8 +81,9 @@ def load_data(self, data_dir, target_path, DATA_READER_CLASS, TARGET_READER_CLAS
             for f in files:
                 fpath = os.path.join(root, f)
 
-                _data = DATA_READER_CLASS(fpath, self._classes, auto_close=True).get_data()
-                _id = _data["_id"]
+                drc = DATA_READER_CLASS(fpath, self._classes, auto_close=True)
+                _data = drc.get_data()
+                _id = drc.get_id()
 
                 if _id in self._data:
                     print("[WARN] Duplicate observation %s found in %s" % (_id, fpath))