[Misc] Update ondisk_dataset_heterograph.ipynb (#7261)

notebooks/stochastic_training/ondisk_dataset_heterograph.ipynb

@@ -411,7 +411,7 @@
         "\n",
         "# Test node paris for user:like:item.\n",
         "lp_test_like_node_pairs_path = os.path.join(base_dir, \"lp-test-like-node-pairs.npy\")\n",
-        "lp_test_like_node_pairs = like_edges[-num_tests, :]\n",
+        "lp_test_like_node_pairs = like_edges[-num_tests:, :]\n",
         "print(f\"Part of test node pairs[user:like:item] for link prediction: {lp_test_like_node_pairs[:3]}\")\n",
         "np.save(lp_test_like_node_pairs_path, lp_test_like_node_pairs)\n",
         "print(f\"LP test node pairs[user:like:item] are saved to {lp_test_like_node_pairs_path}\\n\")\n",
@@ -425,7 +425,7 @@
         "\n",
         "# Test node paris for user:follow:user.\n",
         "lp_test_follow_node_pairs_path = os.path.join(base_dir, \"lp-test-follow-node-pairs.npy\")\n",
-        "lp_test_follow_node_pairs = follow_edges[-num_tests, :]\n",
+        "lp_test_follow_node_pairs = follow_edges[-num_tests:, :]\n",
         "print(f\"Part of test node pairs[user:follow:user] for link prediction: {lp_test_follow_node_pairs[:3]}\")\n",
         "np.save(lp_test_follow_node_pairs_path, lp_test_follow_node_pairs)\n",
         "print(f\"LP test node pairs[user:follow:user] are saved to {lp_test_follow_node_pairs_path}\\n\")\n",
@@ -670,4 +670,4 @@
       "outputs": []
     }
   ]
-}
\ No newline at end of file
+}