the legacy_colorbar matplotlib option doesn't exist in mpl 3.1 or earlier

Author: zingale

advection/simulation.py

@@ -1,7 +1,10 @@
 import importlib
 import numpy as np
 import matplotlib
-matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+try:
+    matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+except KeyError:
+    pass
 import matplotlib.pyplot as plt
 
 import advection.advective_fluxes as flx

advection_nonuniform/simulation.py

@@ -3,7 +3,10 @@
 import importlib
 import numpy as np
 import matplotlib
-matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+try:
+    matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+except KeyError:
+    pass
 import matplotlib.pyplot as plt
 
 import advection_nonuniform.advective_fluxes as flx

compressible/simulation.py

@@ -4,7 +4,10 @@
 
 import numpy as np
 import matplotlib
-matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+try:
+    matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+except KeyError:
+    pass
 import matplotlib.pyplot as plt
 
 import compressible.BC as BC

compressible_react/simulation.py

@@ -1,7 +1,10 @@
 from __future__ import print_function
 
 import matplotlib
-matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+try:
+    matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+except KeyError:
+    pass
 import matplotlib.pyplot as plt
 import numpy as np
 

compressible_sr/simulation.py

@@ -2,7 +2,10 @@
 
 import importlib
 import matplotlib 
-matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+try:
+    matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+except KeyError:
+    pass
 import matplotlib.pyplot as plt
 import numpy as np
 

diffusion/simulation.py

@@ -4,7 +4,10 @@
 import math
 import numpy as np
 import matplotlib
-matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+try:
+    matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+except KeyError:
+    pass
 import matplotlib.pyplot as plt
 
 import mesh.patch as patch

incompressible/simulation.py

@@ -4,7 +4,10 @@
 
 import numpy as np
 import matplotlib
-matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+try:
+    matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+except KeyError:
+    pass
 import matplotlib.pyplot as plt
 
 import incompressible.incomp_interface as incomp_interface

lm_atm/simulation.py

@@ -4,7 +4,10 @@
 
 import numpy as np
 import matplotlib
-matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+try:
+    matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+except KeyError:
+    pass
 import matplotlib.pyplot as plt
 
 import lm_atm.LM_atm_interface as lm_interface

swe/simulation.py

@@ -4,7 +4,10 @@
 
 import numpy as np
 import matplotlib
-matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+try:
+    matplotlib.rcParams['mpl_toolkits.legacy_colorbar'] = False
+except KeyError:
+    pass
 import matplotlib.pyplot as plt
 
 import swe.derives as derives