diff --git a/grum/mainwin.py b/grum/mainwin.py index d0854b0..7fa4758 100644 --- a/grum/mainwin.py +++ b/grum/mainwin.py @@ -2,7 +2,7 @@ from PyQt5.QtCore import Qt, pyqtSignal from PyQt5.QtWidgets import QMainWindow, QSplitter from . import assets -from .descs import PlotDescription, ImageDescription +from .descs import DESC_TYPES, PlotDescription, ImageDescription from .dictlist import DictList from .exampledata import exampledata from .h5filedlg import open_h5_files_dialog, save_h5_file_dialog @@ -208,7 +208,8 @@ class MainWindow(QMainWindow): for fn in fns: data = read_dict(fn) for k, v in data.items(): - DescType = ImageDescription if "image" in v else PlotDescription #TODO + tn = v.pop("type") + DescType = DESC_TYPES[tn] self.add_new_desc_to_list(DescType, k, v)