Question about type of edge_sh in nequip.py

[Hyunsoo Park]

Hi, Thank you for developing this great project.

I am currently working with the nequip.py script and have a question regarding the edge_sh. Given the type of edge_sh is jax.numpy.ndarray, it doesn't seem to have irreps attributes.

Could you check the type of edge_sh ?

Best,