Fixing bug in open source implementation of OGB-LSC/PCQ.

PiperOrigin-RevId: 483319710
ambreshbiradar9 · Oct 24, 2022 · a318c69 · a318c69
1 parent a748a7c
commit a318c69
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ogb_lsc/pcq/dataset_utils.py b/ogb_lsc/pcq/dataset_utils.py
@@ -289,6 +289,7 @@ def _load_smiles(
   elif split == "train":
     indices = datasets.load_all_except_kth_fold_indices(
         data_root, k_fold_split_id, num_k_fold_splits)
+    indices += datasets.load_splits()["train"]
   else:
     assert split == "valid"
     indices = datasets.load_kth_fold_indices(data_root, k_fold_split_id)