properly set option

rapidsai · Aug 8, 2024 · a4ac687 · a4ac687
1 parent 937c9be
commit a4ac687
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/cugraph-pyg/cugraph_pyg/loader/node_loader.py b/python/cugraph-pyg/cugraph_pyg/loader/node_loader.py
@@ -136,6 +136,7 @@ def __init__(
         self.__batch_size = batch_size
         self.__shuffle = shuffle
         self.__drop_last = drop_last
+        self.__global_shuffle = global_shuffle
 
     def __get_input(self):
         _, graph_store = self.__data