Question about type of edge_sh in nequip.py

google-deepmind / materials_discovery

Apache License 2.0

838 stars 131 forks source link

Question about type of edge_sh in nequip.py #11

Open hspark1212 opened 7 months ago

hspark1212 commented 7 months ago

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,