BUG: `setdiff1d` for jax.jit and dask NaN-shaped arrays

Offen

#116 erstellt am 21.1.2025 0 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen

Dieses Issue hat noch niemand übernommen.

Bewertung

Schwierigkeit: 3/5
Geschätzter Aufwand: 1-2 Tage
Anfängerfreundlichkeit: 35/100

Issue-Typ: Bug
Klarheit: Größtenteils klar
Aktivitätsstatus: Veraltet
Tech-Stack: python
Bereich: data

Rechercherichtung

Finde die Implementierung von setdiff1d und ihre xp.unique*-Aufrufe, reproduziere dann zunächst den Fehler innerhalb von jax.jit. Überprüfe das Dask-Backend mit NaN-förmigem x1 und x2 wie beschrieben und füge einen Unit-Test für diesen Fall hinzu. Als erledigt gilt die Aufgabe, wenn setdiff1d für beide gemeldeten Backend-Szenarien funktioniert und die Regression abgedeckt ist.

Vom Indexierungsmodell aus dem Issue-Text verfasst.

Beschreibung

lazy arrays

Follow-up to https://github.com/data-apis/array-api-extra/pull/115

setdiff1d fails when executed inside jax.jit. This is because of a call to xp.unique* without an explicit size= parameter.
By reading the code I'm fairly sure that the function fails if x1 and/or x2 have Dask backend and NaN in their shapes. Lacks a unit test.

Vorherrschende Sprache: Python
Sterne: 33
Forks: 32
Ø Merge: 5 Std. 58 Min.
Gemergte PRs (30 T.): 47

Beitragsleitfaden

Für dieses Repository ist kein Beitragsleitfaden indexiert

Erste Schritte

Lesen Sie das ganze Issue und danach den Beitragsleitfaden des Projekts.
Schreiben Sie ins Issue, dass Sie es übernehmen — das erspart doppelte Arbeit.
Forken Sie das Repository und arbeiten Sie in einem Branch.
Öffnen Sie einen Pull Request, der die Issue-Nummer nennt.

Mehr aus data-apis/array-api-extra

ENH: new function `replace` Offen

enhancement new function

Schwierigkeit 2/5 1-3 Stunden Anfängerfreundlichkeit 72/100

data-apis/array-api-extra#992 · 1 Kommentar · Erstellt am 17.9.2026
TST: `tests-numpy1` windows dtype failures Offen

testing

Schwierigkeit 3/5 1-2 Tage Anfängerfreundlichkeit 55/100

data-apis/array-api-extra#901 · 3 Kommentare · Erstellt am 11.8.2026
TST: multiple nogil `Fatal Python error` observed with `sparse` Offen

testing upstream issue

Schwierigkeit 3/5 1-2 Tage Anfängerfreundlichkeit 45/100

data-apis/array-api-extra#900 · 8 Kommentare · Erstellt am 11.8.2026
TST: non-default device testing Offen

blocked testing xpx.testing

Schwierigkeit 5/5 Über eine Woche Anfängerfreundlichkeit 30/100

data-apis/array-api-extra#833 · 1 Kommentar · 1 Reaktion · Erstellt am 06.7.2026
ENH: pad: implement more padding modes Offen

enhancement help wanted

Schwierigkeit 4/5 3-5 Tage Anfängerfreundlichkeit 55/100

data-apis/array-api-extra#811 · 1 Kommentar · 1 Reaktion · Erstellt am 20.6.2026

Alle Issues in data-apis/array-api-extra