Update types (#564)

Author: dcherian

cf_xarray/accessor.py

@@ -2726,7 +2726,9 @@ def grid_mapping_names(self) -> dict[str, list[str]]:
                 results[v].append(k)
         return results
 
-    def decode_vertical_coords(self, *, outnames=None, prefix=None):
+    def decode_vertical_coords(
+        self, *, outnames: dict[str, str] | None = None, prefix: str | None = None
+    ) -> None:
         """
         Decode parameterized vertical coordinates in place.
 
@@ -2764,6 +2766,7 @@ def decode_vertical_coords(self, *, outnames=None, prefix=None):
         allterms = self.formula_terms
 
         for dim in allterms:
+            dim = cast(str, dim)
             if prefix is None:
                 assert outnames is not None, (
                     "if prefix is None, outnames must be provided"

cf_xarray/parametric.py

@@ -782,7 +782,7 @@ def from_terms(cls, terms: dict):
         return cls(**get_terms(terms, "sigma", "depth", "z1", "z2", "a", "href", "k_c"))
 
 
-TRANSFORM_FROM_STDNAME = {
+TRANSFORM_FROM_STDNAME: dict[str, type[ParametricVerticalCoordinate]] = {
     "atmosphere_ln_pressure_coordinate": AtmosphereLnPressure,
     "atmosphere_sigma_coordinate": AtmosphereSigma,
     "atmosphere_hybrid_sigma_pressure_coordinate": AtmosphereHybridSigmaPressure,