better ignores

scikit-hep · Sep 10, 2024 · 48893b8 · 48893b8
1 parent 1ff29b9
commit 48893b8
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/ragged/_spec_set_functions.py b/src/ragged/_spec_set_functions.py
@@ -126,8 +126,10 @@ def unique_counts(x: array, /) -> tuple[array, array]:
                     values=ragged.array([]), counts=ragged.array([])
                 )
             values, counts = np.unique(
-                x_flat.layout.data, return_counts=True, equal_nan=False
-            )  # pylint: disable=E1101
+                x_flat.layout.data,
+                return_counts=True,
+                equal_nan=False,  # pylint: disable=E1101
+            )
             return unique_counts_result(
                 values=ragged.array(values), counts=ragged.array(counts)
             )
@@ -175,8 +177,10 @@ def unique_inverse(x: array, /) -> tuple[array, array]:
                     values=ragged.array([]), inverse_indices=ragged.array([])
                 )
             values, inverse_indices = np.unique(
-                x_flat.layout.data, return_inverse=True, equal_nan=False
-            )  # pylint: disable=E1101
+                x_flat.layout.data,
+                return_inverse=True,
+                equal_nan=False,  # pylint: disable=E1101
+            )
 
             return unique_inverse_result(
                 values=ragged.array(values),