Fix formatting mistakes moving mypy comments

pydata · Sep 26, 2024 · 700a426 · 700a426
1 parent 33b4576
commit 700a426
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 12 deletions.
diff --git a/xarray/core/groupby.py b/xarray/core/groupby.py
@@ -465,8 +465,9 @@ def factorize(self) -> EncodedGroups:
         # NaNs; as well as values outside the bins are coded by -1
         # Restore these after the raveling
         mask = functools.reduce(
-            np.logical_or, [(code == -1) for code in broadcasted_codes]
-        )  # type: ignore[arg-type]
+            np.logical_or,  # type: ignore[arg-type]
+            [(code == -1) for code in broadcasted_codes],
+        )
         _flatcodes[mask] = -1
 
         midx = pd.MultiIndex.from_product(

diff --git a/xarray/core/resample.py b/xarray/core/resample.py
@@ -188,9 +188,9 @@ def _interpolate(self, kind="linear", **kwargs) -> T_Xarray:
 
 
 # https://github.com/python/mypy/issues/9031
-class DataArrayResample(
+class DataArrayResample(  # type: ignore[misc]
     Resample["DataArray"], DataArrayGroupByBase, DataArrayResampleAggregations
-):  # type: ignore[misc]
+):
     """DataArrayGroupBy object specialized to time resampling operations over a
     specified dimension
     """
@@ -331,9 +331,9 @@ def asfreq(self) -> DataArray:
 
 
 # https://github.com/python/mypy/issues/9031
-class DatasetResample(
+class DatasetResample(  # type: ignore[misc]
     Resample["Dataset"], DatasetGroupByBase, DatasetResampleAggregations
-):  # type: ignore[misc]
+):
     """DatasetGroupBy object specialized to resampling a specified dimension"""
 
     def map(

diff --git a/xarray/namedarray/_array_api.py b/xarray/namedarray/_array_api.py
@@ -79,8 +79,8 @@ def astype(
 
 
 def imag(
-    x: NamedArray[_ShapeType, np.dtype[_SupportsImag[_ScalarType]]],
-    /,  # type: ignore[type-var]
+    x: NamedArray[_ShapeType, np.dtype[_SupportsImag[_ScalarType]]],  # type: ignore[type-var]
+    /,
 ) -> NamedArray[_ShapeType, np.dtype[_ScalarType]]:
     """
     Returns the imaginary component of a complex number for each element x_i of the
@@ -112,8 +112,8 @@ def imag(
 
 
 def real(
-    x: NamedArray[_ShapeType, np.dtype[_SupportsReal[_ScalarType]]],
-    /,  # type: ignore[type-var]
+    x: NamedArray[_ShapeType, np.dtype[_SupportsReal[_ScalarType]]],  # type: ignore[type-var]
+    /,
 ) -> NamedArray[_ShapeType, np.dtype[_ScalarType]]:
     """
     Returns the real component of a complex number for each element x_i of the

diff --git a/xarray/tests/test_coordinates.py b/xarray/tests/test_coordinates.py
@@ -65,8 +65,9 @@ def test_init_index_error(self) -> None:
 
         with pytest.raises(TypeError, match=".* is not an `xarray.indexes.Index`"):
             Coordinates(
-                coords={"x": ("x", [1, 2, 3])}, indexes={"x": "not_an_xarray_index"}
-            )  # type: ignore[dict-item]
+                coords={"x": ("x", [1, 2, 3])},
+                indexes={"x": "not_an_xarray_index"},  # type: ignore[dict-item]
+            )
 
     def test_init_dim_sizes_conflict(self) -> None:
         with pytest.raises(ValueError):