From 1f6ce139e190e5346f7c010e11320a9ed58674f2 Mon Sep 17 00:00:00 2001
From: OscarRoussel <45663923+OscarRoussel@users.noreply.github.com>
Date: Wed, 19 Mar 2025 12:39:53 +0100
Subject: [PATCH] fix loader_pyg.py set_dataset_info

---
 graphgym/loader_pyg.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/graphgym/loader_pyg.py b/graphgym/loader_pyg.py
index 7b7e6735..01bf7124 100644
--- a/graphgym/loader_pyg.py
+++ b/graphgym/loader_pyg.py
@@ -203,11 +203,11 @@ def set_dataset_info(dataset):
 
     # count number of dataset splits
     cfg.share.num_splits = 1
-    for key in dataset.data.keys:
+    for key in dataset.data.keys():
         if 'val' in key:
             cfg.share.num_splits += 1
             break
-    for key in dataset.data.keys:
+    for key in dataset.data.keys():
         if 'test' in key:
             cfg.share.num_splits += 1
             break