Merge pull request #663 from RasmusOrsoe/fix_pyg_import

Change import of `homophily` in `models.utils.py`
graphnet-team · Feb 21, 2024 · 288133c · 288133c
2 parents 2adc935 + 7167bba
commit 288133c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/graphnet/models/utils.py b/src/graphnet/models/utils.py
@@ -6,7 +6,7 @@
 import torch
 from torch import Tensor, LongTensor
 
-from torch_geometric.utils.homophily import homophily
+from torch_geometric.utils import homophily
 
 
 def calculate_xyzt_homophily(