Fix mypy errors

Author: tomvothecoder

e3sm_diags/driver/meridional_mean_2d_driver.py

@@ -82,7 +82,7 @@ def run_diag(parameter: MeridionalMean2dParameter) -> MeridionalMean2dParameter:
                 )
             elif is_vars_3d:
                 if not parameter._is_plevs_set():
-                    parameter.plevs = DEFAULT_PLEVS
+                    parameter.plevs = DEFAULT_PLEVS  # type: ignore
 
                 _run_diags_3d(parameter, ds_test, ds_ref, season, var_key, ref_name)
 

e3sm_diags/parameter/meridional_mean_2d_parameter.py

@@ -10,7 +10,7 @@ def __init__(self):
         super(MeridionalMean2dParameter, self).__init__()
         # Override existing attributes
         # =============================
-        self.plevs = numpy.logspace(2.0, 3.0, num=17).tolist()
+        self.plevs = numpy.logspace(2.0, 3.0, num=17).tolist()  # type: ignore
         self.plot_log_plevs = False
         self.plot_plevs = False
         # Granulating plevs causes duplicate plots in this case.

e3sm_diags/parameter/zonal_mean_2d_parameter.py

@@ -14,7 +14,7 @@ def __init__(self):
         super(ZonalMean2dParameter, self).__init__()
         # Override existing attributes
         # =============================
-        self.plevs = copy.deepcopy(DEFAULT_PLEVS)
+        self.plevs = copy.deepcopy(DEFAULT_PLEVS)  # type: ignore
         self.plot_log_plevs = False
         self.plot_plevs = False
         # Granulating plevs causes duplicate plots in this case.

e3sm_diags/parameter/zonal_mean_2d_stratosphere_parameter.py

@@ -12,5 +12,5 @@ def __init__(self):
         super(ZonalMean2dStratosphereParameter, self).__init__()
         # Override existing attributes
         # =============================
-        self.plevs = copy.deepcopy(DEFAULT_PLEVS)
+        self.plevs = copy.deepcopy(DEFAULT_PLEVS)  # type: ignore
         self.plot_log_plevs = True

e3sm_diags/plot/utils.py

@@ -285,7 +285,7 @@ def _get_colormap(color_map: str):
         raise IOError(msg.format(color_map, pth))
 
     rgb_arr = np.loadtxt(color_map)
-    rgb_arr = rgb_arr / 255.0
+    rgb_arr = rgb_arr / 255.0  # type: ignore
 
     cmap = colors.LinearSegmentedColormap.from_list(name=color_map, colors=rgb_arr)
     return cmap