From 3c90899dfcc4d03e56d21b9d650e2a7e9c0d7ec9 Mon Sep 17 00:00:00 2001
From: Aman Atman <41161981+willtryagain@users.noreply.github.com>
Date: Sat, 11 Feb 2023 11:40:16 +0530
Subject: [PATCH] fix variable name

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 7c12912..85b57d0 100644
--- a/train.py
+++ b/train.py
@@ -55,7 +55,7 @@ def prepare_data(graphs, args, test_graphs=None, max_nodes=0):
     else:
         train_idx = int(len(graphs) * args.train_ratio)
         train_graphs = graphs[:train_idx]
-        val_graphs = graph[train_idx:]
+        val_graphs = graphs[train_idx:]
     print(
         "Num training graphs: ",
         len(train_graphs),