From a806d192c85ef4855b45d9c32fbef961c3490bba Mon Sep 17 00:00:00 2001
From: Grant McConachie <39445771+GrantMcConachie@users.noreply.github.com>
Date: Fri, 6 Dec 2024 09:41:53 -0500
Subject: [PATCH] Update tsagcn.py

Changed to_dense_adj import so it works with the most updated verison of pytorch geometric.
---
 torch_geometric_temporal/nn/attention/tsagcn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_geometric_temporal/nn/attention/tsagcn.py b/torch_geometric_temporal/nn/attention/tsagcn.py
index aa902e2..0f109d7 100644
--- a/torch_geometric_temporal/nn/attention/tsagcn.py
+++ b/torch_geometric_temporal/nn/attention/tsagcn.py
@@ -4,7 +4,7 @@
 import torch.nn as nn
 from torch.autograd import Variable
 try:
-    from torch_geometric.utils.to_dense_adj import to_dense_adj
+    from torch_geometric.utils import to_dense_adj
 except ImportError:
     from torch_geometric.utils import to_dense_adj
 import torch.nn.functional as F