Change import for plots

Author: RJMW

beams/plots.py

@@ -1,22 +1,23 @@
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
 
-from sys import platform
+import sys
 import matplotlib
 
-if "linux" in platform:
+if "linux" in sys.platform or (sys.platform == "darwin" and sys.version_info[0] < 3):
     gui_env = ['TkAgg', 'GTKAgg', 'Qt5Agg', 'WXAgg']
-elif platform == "darwin":
+elif sys.platform == "darwin":
     gui_env = ['Qt5Agg', 'TkAgg', 'GTKAgg', 'WXAgg']
 else:
     pass
 
-if platform != "win32":
+if sys.platform != "win32":
     for gui in gui_env:
         try:
             matplotlib.use(gui, warn=False, force=True)
             break
         except:
+            print(gui)
             continue
 
 import matplotlib.pyplot as plt