More fixes for QtPy transition

glue-viz · Jun 10, 2016 · 7b5eafb · 7b5eafb
1 parent e081a7b
commit 7b5eafb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/glue/core/qt/data_collection_model.py b/glue/core/qt/data_collection_model.py
@@ -492,7 +492,7 @@ def set_data_collection(self, data_collection):
         self._model = DataCollectionModel(data_collection)
         self.setModel(self._model)
 
-        sm = QtCore.QItemSelectionModel(self._model)
+        sm = QtGui.QItemSelectionModel(self._model)
         sm.selectionChanged.connect(lambda *args:
                                     self.selection_changed.emit())
         self.setSelectionModel(sm)

diff --git a/glue/viewers/table/qt/viewer_widget.py b/glue/viewers/table/qt/viewer_widget.py
@@ -162,7 +162,7 @@ def _update_selection(self):
             index = self.model.order[index]
             model_index = self.model.createIndex(index, 0)
             model.select(model_index,
-                         QtCore.QItemSelectionModel.Select | QtCore.QItemSelectionModel.Rows)
+                         QtGui.QItemSelectionModel.Select | QtGui.QItemSelectionModel.Rows)
 
         self.ui.table.setSelectionMode(selection_mode)