update

pyg-team · Oct 12, 2023 · 450980f · 450980f
1 parent fa0048d
commit 450980f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/datasets/test_torch_frame_datasets.py b/test/datasets/test_torch_frame_datasets.py
@@ -10,7 +10,9 @@
     TaskType.REGRESSION
 ])
 def test_dataframe_benchmark_match(task_type, scale):
-    # Make sure task_type, scale, idx triple map to the same underlying dataset
+    # Make sure task_type, scale, idx triple map to the fixed underlying
+    # dataset. New dataset can be appeneded, but the existing mapping needes to
+    # be preserved.
     datasets = DataFrameBenchmark.datasets_available(task_type=task_type,
                                                      scale=scale)
     if task_type == TaskType.BINARY_CLASSIFICATION: