[gui] try to fix matplotlib backend for Qt5

pymor · Apr 27, 2017 · c2ab02d · c2ab02d
1 parent ce0adf2
commit c2ab02d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/pymor/gui/matplotlib.py b/src/pymor/gui/matplotlib.py
@@ -17,7 +17,13 @@
 if config.HAVE_QT and config.HAVE_MATPLOTLIB:
     from Qt.QtWidgets import QSizePolicy
 
-    from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
+    import Qt
+    if Qt.__qt_version__[0] == '4':
+        from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
+    elif Qt.__qt_version__[0] == '5':
+        from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
+    else:
+        raise NotImplementedError
 
     from matplotlib.figure import Figure