Update filtered list for rft healthbench

Author: robtinn

examples/fine-tuned_qa/reinforcement_finetuning_healthbench.ipynb

@@ -144,7 +144,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 19,
+   "execution_count": 4,
    "id": "4f02e651",
    "metadata": {},
    "outputs": [],
@@ -191,7 +191,7 @@
     "\n",
     "# Only include examples that have been pre-filtered to make the RFT job simple to run and evaluate\n",
     "filter_indices = set(\n",
-    "    [0, 1, 2, 7, 8, 9, 10, 12, 15, 20, 21, 25, 26, 27, 30, 33, 35, 38, 39, 41, 44, 45, 47, 49, 50]\n",
+    "    [0, 1, 2, 7, 8, 9, 10, 12, 15, 20, 21, 26, 27, 30, 35, 38, 39, 41, 44, 45, 47, 49, 50]\n",
     ")\n",
     "filtered_data = []\n",
     "for i, datapoint in enumerate(data):\n",
@@ -272,15 +272,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 18,
    "id": "07c3f15f",
    "metadata": {},
    "outputs": [],
    "source": [
     "# Split data\n",
     "train_datapoints = [{\"messages\": data[\"prompt\"][1:]} for data in filtered_data][:12]\n",
-    "val_datapoints = [{\"messages\": data[\"prompt\"][1:]} for data in filtered_data][12:22]\n",
-    "test_datapoints = [{\"messages\": data[\"prompt\"][1:], \"completion\": data[\"completion\"]} for data in filtered_data][22:]\n",
+    "val_datapoints = [{\"messages\": data[\"prompt\"][1:]} for data in filtered_data][12:20]\n",
+    "test_datapoints = [{\"messages\": data[\"prompt\"][1:], \"completion\": data[\"completion\"]} for data in filtered_data][20:]\n",
     "\n",
     "# Write to files\n",
     "train_path = 'local_cache/rft_train.jsonl'\n",