matplotlib needs Qt5Agg backend in Ubuntu 16.04

ahundt · Jan 2, 2017 · 17220c0 · 17220c0
1 parent 3bae055
commit 17220c0
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/python/gps/gps_main.py b/python/gps/gps_main.py
@@ -1,7 +1,20 @@
 """ This file defines the main object that runs experiments. """
 
 import matplotlib as mpl
-mpl.use('Qt4Agg')
+qt_found=False
+try:
+    import PyQt4
+    mpl.use('Qt4Agg')
+    qt_found = True
+except ImportError:
+    qt_found = False
+if qt_found == False:
+    try:
+        import PyQt5
+        mpl.use('Qt5Agg')
+        qt_found = True
+    except ImportError:
+        qt_found = False
 
 import logging
 import imp