fix: backport vendored mixin from #2534#2583
Conversation
| ) | ||
|
|
||
| elif ak.nplike.is_jax_buffer(array) and type(array).__name__ == "DeviceArray": | ||
| elif ak.nplike.is_jax_buffer(array): |
There was a problem hiding this comment.
Any jaxlib object is now permitted here, which mirrors v2.
jpivarski
left a comment
There was a problem hiding this comment.
Why is there a _v2 directory? I noticed that NDArrayOperatorsMixin is now defined twice, and then one of them is in the _v2 directory.
This is a good thing to do, though!
|
@jpivarski this is a backport for |
Codecov Report
Additional details and impacted files |
This PR uses only the vendored
NDArrayOperatorsMixinmixin class (without__slots__) from #2534