From 7167bba5e86df72af434def80b836934050b744c Mon Sep 17 00:00:00 2001 From: Rasmus Oersoe Date: Tue, 20 Feb 2024 14:23:05 +0100 Subject: [PATCH] change import of `homophily` --- src/graphnet/models/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/graphnet/models/utils.py b/src/graphnet/models/utils.py index e1ef7956c..1d77903a5 100644 --- 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(