glass icon indicating copy to clipboard operation
glass copied to clipboard

`fields.effective_cls` is still calling np

Open connoraird opened this issue 5 months ago • 1 comments

Describe the Bug

Although fields.effective_cls was ported in #407 np is still being called directly...

pairs = (
    combinations_with_replacement(np.ndindex(shape1[1:]), 2)
    if weights2 is weights1
    else product(np.ndindex(shape1[1:]), np.ndindex(shape2[1:]))
)

To Reproduce


Expected Behaviour

We should not be calling np but instead one of xp, uxpx or xpx

Actual Behaviour

No response

Version In Use

main

Additional Context


connoraird avatar Nov 20 '25 12:11 connoraird

Seems this is blocked by ndindex not being in the Array API

paddyroddy avatar Nov 20 '25 16:11 paddyroddy