Fix mypy issues

E3SM-Project · Jan 21, 2025 · ef060ee · ef060ee
1 parent 72a1d0b
commit ef060ee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/e3sm_diags/driver/arm_diags_driver.py b/e3sm_diags/driver/arm_diags_driver.py
@@ -161,7 +161,7 @@ def _run_diag_diurnal_cycle_zt(parameter: ARMDiagsParameter) -> ARMDiagsParamete
     ref_path = parameter.reference_data_path
 
     seasons = SEASONS_BY_DIAG["diurnal_cycle_zt"]
-    plevs = list(np.linspace(100, 1000, 37))
+    plevs: List[float] = list(np.linspace(100, 1000, 37))
 
     for region in regions:
         logger.info(f"Selected region: {region}")

diff --git a/tests/e3sm_diags/driver/utils/test_dataset_xr.py b/tests/e3sm_diags/driver/utils/test_dataset_xr.py
@@ -241,7 +241,8 @@ def test_sets_drived_vars_map_with_existing_entry(self):
         # The expected `derived_vars_map` result.
         expected = DERIVED_VARIABLES.copy()
         expected["PRECT"] = OrderedDict(
-            **parameter.derived_variables["PRECT"], **expected["PRECT"]
+            **parameter.derived_variables["PRECT"],
+            **expected["PRECT"],  # type: ignore
         )
 
         assert ds.derived_vars_map == expected