search

graphnet-team / graphnet

A Deep learning library for neutrino telescopes
https://graphnet-team.github.io/graphnet/
Apache License 2.0
85 stars 86 forks source link

Change import of `homophily` in `models.utils.py` #663

Closed RasmusOrsoe closed 4 months ago

RasmusOrsoe commented 4 months ago

In a recent change in PyG, the module torch_geometric.utils.homophily was re-named to torc_geometric.utils._homophily and as a result, the import statement from torch_geometric.utils.homophily import homophily in graphnet.models.utils.py fails. This has caused recent checks to fails on PRs, see #662.

This PR changes the import statement

from torch_geometric.utils.homophily import homophily

to

from torch_geometric.utils import homophily

This change is backwards compatible.

RasmusOrsoe commented 4 months ago

@Aske-Rosted this should solve the issue