Merge pull request #271 from c0mm4nd/master

Fix ModuleNotFoundError: No module named 'torch_geometric.utils.to_dense_adj'
benedekrozemberczki · Oct 14, 2024 · b2833be · b2833be
2 parents 9d1b5ab + 59bf3e6
commit b2833be
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/torch_geometric_temporal/nn/attention/tsagcn.py b/torch_geometric_temporal/nn/attention/tsagcn.py
@@ -3,7 +3,10 @@
 import numpy as np
 import torch.nn as nn
 from torch.autograd import Variable
-from torch_geometric.utils.to_dense_adj import to_dense_adj
+try:
+    from torch_geometric.utils.to_dense_adj import to_dense_adj
+except ImportError:
+    from torch_geometric.utils import to_dense_adj
 import torch.nn.functional as F