google-deepmind / google-deepmind/materials_discovery
Question about type of edge_sh in nequip.py
Open
- Dominant language
- Jupyter Notebook
- Stars
- 1.2k
- Forks
- 197
- PR merge metrics
- No merged PRs in 30d
Description
Hi, Thank you for developing this great project.
I am currently working with the [nequip.py](https://github.com/google-deepmind/materials_discovery/blob/main/model/nequip.py) script and have a question regarding the [`edge_sh`](https://github.com/google-deepmind/materials_discovery/blob/7372c3a2f90d480e62d61ec386451a321045cea7/model/nequip.py#L138C35-L138C35). 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,
Contributor guide
Assessment
This issue has not been assessed yet.