ENH: setdiff1d: add delegation

[Enderdead]

This merge request introduces the setdiff1d delegate function (#100).

Nothing special to mention here — it was NOT a straightforward contribution.

[lucascolley]

see also #124

[Enderdead]

Two issues have been identified during the development of the JAX part (see jax-ml/jax#32335 and jax-ml/jax#32402 )

I will finish my contribution once all the issues have been reported to the JAX library.

[Enderdead]

According to my understanding of the work discussed in #124, the JAX JIT version has been postponed indefinitely, but the Dask version is still desired. Since the function setdiff1d is not available in the Dask API (see Dask NumPy compatibility), the implementation proposed by @crusaderky seems worth pushing forward.

Because this implementation differs from a classic delegate function (as it introduces a new approach), where should I place the dedicated code for Dask? Alternatively, do we want to replace the current version, which uses the _helpers.in1d method, with @crusaderky’s implementation?

[lucascolley]

I'm not sure about how to proceed for Dask and JAX, but I would be happy to review a PR which just adds delegation for the other libraries.

[lucascolley]

LGTM, thanks @Enderdead !

[crusaderky]

Just a nit on the tests.

I checked that the documentation builds correctly.

[Enderdead]

@crusaderky Corrections made to the tests.

[lucascolley]

Did you mean to include a change to atleast_nd too?

[Enderdead]

@lucascolley I swapped the function order in the file to follow alphabetical order and added the missing item to the __all__ list. Is that a problem? I can split this into a second commit for clarity.

[lucascolley]

all good, thanks again @Enderdead ! And for the review @crusaderky