From 482ddd67a68fbe2f697bf9c5d8992a2120230f23 Mon Sep 17 00:00:00 2001 From: araison Date: Fri, 30 Dec 2022 19:47:07 +0100 Subject: [PATCH] Fixing bug --- explaining_framework/metric/fidelity.py | 1 + 1 file changed, 1 insertion(+) diff --git a/explaining_framework/metric/fidelity.py b/explaining_framework/metric/fidelity.py index c80cccb..96cf944 100644 --- a/explaining_framework/metric/fidelity.py +++ b/explaining_framework/metric/fidelity.py @@ -3,6 +3,7 @@ import torch.nn.functional as F from torch.nn import KLDivLoss, Softmax from torch_geometric.explain.explanation import Explanation from torch_geometric.graphgym.config import cfg +from torch import Tensor from explaining_framework.metric.base import Metric