From 48893b8bba2036d21ba4c3d5bf67b0b33d3b00f9 Mon Sep 17 00:00:00 2001 From: ohrechykha Date: Tue, 10 Sep 2024 15:12:28 +0300 Subject: [PATCH] better ignores --- src/ragged/_spec_set_functions.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) 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),