fix: error when no train_mask in dgl

Author: galogm

.ci/install-cpu.sh

@@ -1,7 +1,7 @@
 # # install python 3.8.16 use pyenv
 # sudo apt-get install zlib1g-dev libffi-dev libreadline-dev libssl-dev libsqlite3-dev libncurses5 libncurses5-dev libncursesw5 lzma liblzma-dev libbz2-dev
 # pyenv install 3.8.16
-# pyenv local 38.16
+# pyenv local 3.8.16
 
 # create and activate virtual environment
 if [ ! -d '.env' ]; then

graph_datasets/datasets/pyg.py

@@ -87,10 +87,6 @@ def load_pyg_data(
         graph.ndata["train_mask"] = data.train_mask
         graph.ndata["val_mask"] = data.val_mask
         graph.ndata["test_mask"] = data.test_mask
-    else:
-        graph.ndata["train_mask"] = None
-        graph.ndata["val_mask"] = None
-        graph.ndata["test_mask"] = None
 
     if verbosity and verbosity > 1:
         print_dataset_info(