diff --git a/sparse/_coo/common.py b/sparse/_coo/common.py index 9de529be..b99a78b4 100644 --- a/sparse/_coo/common.py +++ b/sparse/_coo/common.py @@ -1061,6 +1061,7 @@ def clip(a, a_min=None, a_max=None, out=None): # Array API set functions + class UniqueAllResult(NamedTuple): values: np.ndarray indices: np.ndarray diff --git a/sparse/tests/test_coo.py b/sparse/tests/test_coo.py index 9f579362..faaeff11 100644 --- a/sparse/tests/test_coo.py +++ b/sparse/tests/test_coo.py @@ -1796,13 +1796,7 @@ def test_unique_values(self, arr): np.testing.assert_equal(result, expected) @pytest.mark.parametrize( - "func", - [ - sparse.unique_all, - sparse.unique_counts, - sparse.unique_inverse, - sparse.unique_values - ] + "func", [sparse.unique_all, sparse.unique_counts, sparse.unique_inverse, sparse.unique_values] ) def test_input_validation(self, func): with pytest.raises(ValueError, match=r"Only COO arrays are supported"):