fix bugs

gdmnl · Apr 3, 2024 · e31f417 · e31f417
2 parents b06bd98 + 771ed05
commit e31f417
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyg_spectral/nn/conv/bern_conv.py b/pyg_spectral/nn/conv/bern_conv.py
@@ -9,7 +9,7 @@
 from torch_scatter import scatter_add
 from torch_sparse import SparseTensor, matmul, fill_diag, sum, mul
 from torch_geometric.nn.conv import MessagePassing
-from torch_geometric.utils import add_self_loops, remove_self_loops #,get_laplacian
+from torch_geometric.utils import spmm, add_self_loops, remove_self_loops #,get_laplacian
 from pyg_spectral.utils import get_laplacian
 from torch_geometric.utils.num_nodes import maybe_num_nodes
 from torch_geometric.nn.conv.gcn_conv import gcn_norm