Restrict cached loader option to homogenous models

pyg-team · Aug 17, 2023 · acf12f3 · acf12f3
1 parent 76c0c14
commit acf12f3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/benchmark/inference/inference_benchmark.py b/benchmark/inference/inference_benchmark.py
@@ -80,6 +80,9 @@ def run(args: argparse.ArgumentParser):
         _, _, test_mask = get_split_masks(data, dataset_name)
         degree = None
 
+        if hetero and args.cached_loader:
+            args.cached_loader = False
+            print('Disabling CachedLoader, not supported in Hetero models')
         if args.num_layers != [1] and not hetero and args.num_steps != -1:
             raise ValueError("Layer-wise inference requires `steps=-1`")