diff --git a/src/ragged/_spec_set_functions.py b/src/ragged/_spec_set_functions.py index ed0a2db..326edf9 100644 --- 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),