|
18 | 18 | Any,
|
19 | 19 | Literal,
|
20 | 20 | TypeVar,
|
21 |
| - Union, |
22 | 21 | cast,
|
23 | 22 | overload,
|
24 | 23 | )
|
|
89 | 88 | ATTRS["vertical"] = ATTRS["Z"]
|
90 | 89 |
|
91 | 90 | # Type for Mapper functions
|
92 |
| -Mapper = Callable[[Union[DataArray, Dataset], Hashable], list[Hashable]] |
| 91 | +Mapper = Callable[[DataArray | Dataset, Hashable], list[Hashable]] |
93 | 92 |
|
94 | 93 | # Type for decorators
|
95 | 94 | F = TypeVar("F", bound=Callable[..., Any])
|
@@ -1150,9 +1149,10 @@ def create_flag_dict(da) -> Mapping[Hashable, FlagParam]:
|
1150 | 1149 | )
|
1151 | 1150 |
|
1152 | 1151 | flag_params = tuple(
|
1153 |
| - FlagParam(mask, value) for mask, value in zip(flag_masks, flag_values) |
| 1152 | + FlagParam(mask, value) |
| 1153 | + for mask, value in zip(flag_masks, flag_values, strict=False) |
1154 | 1154 | )
|
1155 |
| - return dict(zip(flag_meanings, flag_params)) |
| 1155 | + return dict(zip(flag_meanings, flag_params, strict=False)) |
1156 | 1156 |
|
1157 | 1157 |
|
1158 | 1158 | class CFAccessor:
|
@@ -1368,7 +1368,7 @@ def curvefit(
|
1368 | 1368 | kwargs: dict[str, Any] | None = None,
|
1369 | 1369 | ):
|
1370 | 1370 | if coords is not None:
|
1371 |
| - if isinstance(coords, (Hashable, DataArray)): |
| 1371 | + if isinstance(coords, Hashable | DataArray): |
1372 | 1372 | coords_iter: Iterable[Hashable | DataArray] = [coords]
|
1373 | 1373 | else:
|
1374 | 1374 | coords_iter = coords
|
@@ -3024,7 +3024,7 @@ def _extract_flags(self, flags: Sequence[Hashable] | None = None) -> Dataset:
|
3024 | 3024 | x = self._obj.astype("i")
|
3025 | 3025 | bit_comp = x & bit_mask
|
3026 | 3026 |
|
3027 |
| - for i, (flag, value) in enumerate(zip(flags_reduced, values)): |
| 3027 | + for i, (flag, value) in enumerate(zip(flags_reduced, values, strict=False)): |
3028 | 3028 | bit = bit_comp.isel(_mask=i)
|
3029 | 3029 | if value is not None:
|
3030 | 3030 | out[flag] = bit == value
|
|
0 commit comments