diff --git a/.gitignore b/.gitignore
index 7fef19b3..359a8975 100644
--- a/.gitignore
+++ b/.gitignore
@@ -5,3 +5,4 @@ __pycache__
 .idea/
 venv/
 results/
+cache
diff --git a/tutorials/custom_dataset.ipynb b/tutorials/custom_dataset.ipynb
index 368361db..8e52c360 100644
--- a/tutorials/custom_dataset.ipynb
+++ b/tutorials/custom_dataset.ipynb
@@ -48,15 +48,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 16,
+   "execution_count": 1,
    "id": "e78e8039-1bb8-4265-ac9e-65520deab3a5",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:12:28.637184Z",
-     "iopub.status.busy": "2024-07-25T05:12:28.636694Z",
-     "iopub.status.idle": "2024-07-25T05:12:28.671479Z",
-     "shell.execute_reply": "2024-07-25T05:12:28.670310Z",
-     "shell.execute_reply.started": "2024-07-25T05:12:28.637140Z"
+     "iopub.execute_input": "2024-07-26T00:25:19.877872Z",
+     "iopub.status.busy": "2024-07-26T00:25:19.877382Z",
+     "iopub.status.idle": "2024-07-26T00:25:20.526819Z",
+     "shell.execute_reply": "2024-07-26T00:25:20.525898Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:19.877832Z"
     }
    },
    "outputs": [],
@@ -202,11 +202,11 @@
    "id": "165d2698-ae4c-44f9-be75-24cf55028bde",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T04:45:09.974607Z",
-     "iopub.status.busy": "2024-07-25T04:45:09.974267Z",
-     "iopub.status.idle": "2024-07-25T04:45:10.004229Z",
-     "shell.execute_reply": "2024-07-25T04:45:10.003658Z",
-     "shell.execute_reply.started": "2024-07-25T04:45:09.974590Z"
+     "iopub.execute_input": "2024-07-26T00:25:22.150312Z",
+     "iopub.status.busy": "2024-07-26T00:25:22.149838Z",
+     "iopub.status.idle": "2024-07-26T00:25:22.180445Z",
+     "shell.execute_reply": "2024-07-26T00:25:22.179540Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:22.150280Z"
     }
    },
    "outputs": [],
@@ -492,15 +492,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 28,
+   "execution_count": 3,
    "id": "65c18113-7067-41cf-81da-bb159f17e581",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:22:52.837922Z",
-     "iopub.status.busy": "2024-07-25T05:22:52.837464Z",
-     "iopub.status.idle": "2024-07-25T05:22:52.869823Z",
-     "shell.execute_reply": "2024-07-25T05:22:52.869269Z",
-     "shell.execute_reply.started": "2024-07-25T05:22:52.837884Z"
+     "iopub.execute_input": "2024-07-26T00:25:23.545040Z",
+     "iopub.status.busy": "2024-07-26T00:25:23.544759Z",
+     "iopub.status.idle": "2024-07-26T00:25:23.575626Z",
+     "shell.execute_reply": "2024-07-26T00:25:23.574925Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:23.545014Z"
     }
    },
    "outputs": [
@@ -510,27 +510,27 @@
        "F1Dataset()"
       ]
      },
-     "execution_count": 28,
+     "execution_count": 3,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
-    "f1_dataset = F1Dataset(cache_dir=\"~/.cache/relbench_custom/f1\")\n",
+    "f1_dataset = F1Dataset(cache_dir=\"./cache/f1\")\n",
     "f1_dataset"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 4,
    "id": "7d9de641-13ae-442e-bfa0-08d5efa5a1c1",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:01:22.854398Z",
-     "iopub.status.busy": "2024-07-25T05:01:22.854078Z",
-     "iopub.status.idle": "2024-07-25T05:01:23.928765Z",
-     "shell.execute_reply": "2024-07-25T05:01:23.928198Z",
-     "shell.execute_reply.started": "2024-07-25T05:01:22.854378Z"
+     "iopub.execute_input": "2024-07-26T00:25:24.277335Z",
+     "iopub.status.busy": "2024-07-26T00:25:24.276960Z",
+     "iopub.status.idle": "2024-07-26T00:25:24.603887Z",
+     "shell.execute_reply": "2024-07-26T00:25:24.603241Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:24.277301Z"
     }
    },
    "outputs": [
@@ -540,8 +540,8 @@
      "text": [
       "Making Database object from scratch...\n",
       "(You can also use `get_dataset(..., download=True)` for datasets prepared by the RelBench team.)\n",
-      "Done in 0.23 seconds.\n",
-      "Caching Database object to ./cache/custom-f1/db...\n",
+      "Done in 0.24 seconds.\n",
+      "Caching Database object to ./cache/f1/db...\n",
       "Done in 0.05 seconds.\n"
      ]
     },
@@ -551,7 +551,7 @@
        "Database()"
       ]
      },
-     "execution_count": 8,
+     "execution_count": 4,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -563,15 +563,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 5,
    "id": "6ab7a185-fc58-48fd-a8a9-7b38b265acb3",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:02:06.500844Z",
-     "iopub.status.busy": "2024-07-25T05:02:06.500540Z",
-     "iopub.status.idle": "2024-07-25T05:02:06.532747Z",
-     "shell.execute_reply": "2024-07-25T05:02:06.532190Z",
-     "shell.execute_reply.started": "2024-07-25T05:02:06.500810Z"
+     "iopub.execute_input": "2024-07-26T00:25:24.813771Z",
+     "iopub.status.busy": "2024-07-26T00:25:24.813564Z",
+     "iopub.status.idle": "2024-07-26T00:25:24.864693Z",
+     "shell.execute_reply": "2024-07-26T00:25:24.863584Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:24.813754Z"
     }
    },
    "outputs": [
@@ -611,7 +611,7 @@
        "  time_col=date)"
       ]
      },
-     "execution_count": 10,
+     "execution_count": 5,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -638,15 +638,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 6,
    "id": "53f7fa8e-ec89-4bc5-b8ef-1037e0c25917",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:06:39.656930Z",
-     "iopub.status.busy": "2024-07-25T05:06:39.656575Z",
-     "iopub.status.idle": "2024-07-25T05:06:39.881534Z",
-     "shell.execute_reply": "2024-07-25T05:06:39.880677Z",
-     "shell.execute_reply.started": "2024-07-25T05:06:39.656909Z"
+     "iopub.execute_input": "2024-07-26T00:25:26.487123Z",
+     "iopub.status.busy": "2024-07-26T00:25:26.486671Z",
+     "iopub.status.idle": "2024-07-26T00:25:26.720974Z",
+     "shell.execute_reply": "2024-07-26T00:25:26.719924Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:26.487085Z"
     }
    },
    "outputs": [
@@ -656,7 +656,7 @@
        "Database()"
       ]
      },
-     "execution_count": 11,
+     "execution_count": 6,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -668,15 +668,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 7,
    "id": "942649f9-4527-4dba-817f-52eb44851189",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:07:11.745448Z",
-     "iopub.status.busy": "2024-07-25T05:07:11.744850Z",
-     "iopub.status.idle": "2024-07-25T05:07:11.766078Z",
-     "shell.execute_reply": "2024-07-25T05:07:11.765549Z",
-     "shell.execute_reply.started": "2024-07-25T05:07:11.745429Z"
+     "iopub.execute_input": "2024-07-26T00:25:27.079812Z",
+     "iopub.status.busy": "2024-07-26T00:25:27.079571Z",
+     "iopub.status.idle": "2024-07-26T00:25:27.106195Z",
+     "shell.execute_reply": "2024-07-26T00:25:27.105351Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:27.079795Z"
     }
    },
    "outputs": [
@@ -716,7 +716,7 @@
        "  time_col=date)"
       ]
      },
-     "execution_count": 12,
+     "execution_count": 7,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -743,15 +743,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": 8,
    "id": "3df1be1a-3024-4cec-9ef7-17b645af6aa5",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:12:34.400469Z",
-     "iopub.status.busy": "2024-07-25T05:12:34.400200Z",
-     "iopub.status.idle": "2024-07-25T05:12:34.418477Z",
-     "shell.execute_reply": "2024-07-25T05:12:34.417728Z",
-     "shell.execute_reply.started": "2024-07-25T05:12:34.400451Z"
+     "iopub.execute_input": "2024-07-26T00:25:28.421274Z",
+     "iopub.status.busy": "2024-07-26T00:25:28.420889Z",
+     "iopub.status.idle": "2024-07-26T00:25:28.448827Z",
+     "shell.execute_reply": "2024-07-26T00:25:28.447787Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:28.421239Z"
     }
    },
    "outputs": [
@@ -768,7 +768,7 @@
        " 'rel-custom_f1']"
       ]
      },
-     "execution_count": 17,
+     "execution_count": 8,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -780,15 +780,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 18,
+   "execution_count": 9,
    "id": "ca0fe893-b439-4f7f-aff1-f7e19200ecf9",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:13:08.883061Z",
-     "iopub.status.busy": "2024-07-25T05:13:08.882777Z",
-     "iopub.status.idle": "2024-07-25T05:13:08.913285Z",
-     "shell.execute_reply": "2024-07-25T05:13:08.912316Z",
-     "shell.execute_reply.started": "2024-07-25T05:13:08.883041Z"
+     "iopub.execute_input": "2024-07-26T00:25:28.943803Z",
+     "iopub.status.busy": "2024-07-26T00:25:28.943378Z",
+     "iopub.status.idle": "2024-07-26T00:25:28.973952Z",
+     "shell.execute_reply": "2024-07-26T00:25:28.973012Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:28.943766Z"
     }
    },
    "outputs": [
@@ -798,7 +798,7 @@
        "F1Dataset()"
       ]
      },
-     "execution_count": 18,
+     "execution_count": 9,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -842,15 +842,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 19,
+   "execution_count": 10,
    "id": "851458c3-c9bf-48b5-a09e-f5a802ef2e62",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:14:10.259967Z",
-     "iopub.status.busy": "2024-07-25T05:14:10.259483Z",
-     "iopub.status.idle": "2024-07-25T05:14:10.318804Z",
-     "shell.execute_reply": "2024-07-25T05:14:10.317667Z",
-     "shell.execute_reply.started": "2024-07-25T05:14:10.259924Z"
+     "iopub.execute_input": "2024-07-26T00:25:30.982702Z",
+     "iopub.status.busy": "2024-07-26T00:25:30.982290Z",
+     "iopub.status.idle": "2024-07-26T00:25:31.037729Z",
+     "shell.execute_reply": "2024-07-26T00:25:31.036603Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:30.982666Z"
     }
    },
    "outputs": [],
@@ -1085,15 +1085,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 22,
+   "execution_count": 11,
    "id": "c315c96a-5450-47d8-b974-c1f61510366b",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:17:06.453379Z",
-     "iopub.status.busy": "2024-07-25T05:17:06.452856Z",
-     "iopub.status.idle": "2024-07-25T05:17:06.486342Z",
-     "shell.execute_reply": "2024-07-25T05:17:06.485324Z",
-     "shell.execute_reply.started": "2024-07-25T05:17:06.453341Z"
+     "iopub.execute_input": "2024-07-26T00:25:31.624395Z",
+     "iopub.status.busy": "2024-07-26T00:25:31.624017Z",
+     "iopub.status.idle": "2024-07-26T00:25:31.654247Z",
+     "shell.execute_reply": "2024-07-26T00:25:31.653223Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:31.624360Z"
     }
    },
    "outputs": [
@@ -1103,7 +1103,7 @@
        "AmazonDataset()"
       ]
      },
-     "execution_count": 22,
+     "execution_count": 11,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1115,25 +1115,50 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 24,
+   "execution_count": 12,
    "id": "bc68cd58-d4fa-4433-a95b-6b912167b761",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:17:41.674551Z",
-     "iopub.status.busy": "2024-07-25T05:17:41.674267Z",
-     "iopub.status.idle": "2024-07-25T05:17:41.698677Z",
-     "shell.execute_reply": "2024-07-25T05:17:41.697675Z",
-     "shell.execute_reply.started": "2024-07-25T05:17:41.674533Z"
+     "iopub.execute_input": "2024-07-26T00:25:31.797241Z",
+     "iopub.status.busy": "2024-07-26T00:25:31.796793Z",
+     "iopub.status.idle": "2024-07-26T00:25:32.211913Z",
+     "shell.execute_reply": "2024-07-26T00:25:32.210878Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:31.797192Z"
     }
    },
    "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Making Database object from scratch...\n",
+      "(You can also use `get_dataset(..., download=True)` for datasets prepared by the RelBench team.)\n",
+      "reading product info from /lfs/ampere4/0/ranjanr/.cache/pooch/b70e8d295f37e2465ea17803b6d1e11d-meta_AMAZON_FASHION.json.gz.decomp...\n",
+      "done in 0.07 seconds.\n",
+      "converting to pandas dataframe...\n",
+      "done in 0.16 seconds.\n",
+      "processing product info...\n",
+      "done in 0.11 seconds.\n",
+      "reading review and customer info from /lfs/ampere4/0/ranjanr/.cache/pooch/26323778935ec86761e2c260cea27601-AMAZON_FASHION_5.json.gz.decomp...\n",
+      "done in 0.01 seconds.\n",
+      "converting to pandas dataframe...\n",
+      "done in 0.00 seconds.\n",
+      "processing review and customer info...\n",
+      "done in 0.00 seconds.\n",
+      "keeping only products common to product and review tables...\n",
+      "done in 0.01 seconds.\n",
+      "extracting customer table...\n",
+      "done in 0.00 seconds.\n",
+      "Done in 0.38 seconds.\n"
+     ]
+    },
     {
      "data": {
       "text/plain": [
        "Database()"
       ]
      },
-     "execution_count": 24,
+     "execution_count": 12,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1153,15 +1178,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 26,
+   "execution_count": 13,
    "id": "92e0a135-3fb2-4ff4-8e9a-9964e8f037e5",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:19:22.570729Z",
-     "iopub.status.busy": "2024-07-25T05:19:22.570282Z",
-     "iopub.status.idle": "2024-07-25T05:19:22.602399Z",
-     "shell.execute_reply": "2024-07-25T05:19:22.601289Z",
-     "shell.execute_reply.started": "2024-07-25T05:19:22.570693Z"
+     "iopub.execute_input": "2024-07-26T00:25:32.213317Z",
+     "iopub.status.busy": "2024-07-26T00:25:32.213145Z",
+     "iopub.status.idle": "2024-07-26T00:25:32.231430Z",
+     "shell.execute_reply": "2024-07-26T00:25:32.230569Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:32.213302Z"
     }
    },
    "outputs": [
@@ -1179,7 +1204,7 @@
        " 'rel-amazon_fashion']"
       ]
      },
-     "execution_count": 26,
+     "execution_count": 13,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1193,15 +1218,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 27,
+   "execution_count": 14,
    "id": "9dda56b3-43d6-43fb-b28a-cfc172db26eb",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T05:19:49.467811Z",
-     "iopub.status.busy": "2024-07-25T05:19:49.467395Z",
-     "iopub.status.idle": "2024-07-25T05:19:49.488490Z",
-     "shell.execute_reply": "2024-07-25T05:19:49.487904Z",
-     "shell.execute_reply.started": "2024-07-25T05:19:49.467789Z"
+     "iopub.execute_input": "2024-07-26T00:25:32.375618Z",
+     "iopub.status.busy": "2024-07-26T00:25:32.375348Z",
+     "iopub.status.idle": "2024-07-26T00:25:32.392944Z",
+     "shell.execute_reply": "2024-07-26T00:25:32.392057Z",
+     "shell.execute_reply.started": "2024-07-26T00:25:32.375595Z"
     }
    },
    "outputs": [
@@ -1211,7 +1236,7 @@
        "AmazonDataset()"
       ]
      },
-     "execution_count": 27,
+     "execution_count": 14,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -1242,7 +1267,7 @@
    "id": "10a22967-f376-4157-be22-d4443c04d108",
    "metadata": {},
    "source": [
-    "To use the custom dataset within the RelBench framework, you would want to define custom tasks on it. See this tutorial for how to do that: [../custom_task.ipynb](../custom_task.ipynb)"
+    "To use the custom dataset within the RelBench framework, you would want to define custom tasks on it. See this tutorial for how to do that: [custom_task.ipynb](custom_task.ipynb)"
    ]
   },
   {
diff --git a/tutorials/custom_task.ipynb b/tutorials/custom_task.ipynb
index a7f0f691..8d1b8c88 100644
--- a/tutorials/custom_task.ipynb
+++ b/tutorials/custom_task.ipynb
@@ -58,15 +58,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 15,
+   "execution_count": 1,
    "id": "e78e8039-1bb8-4265-ac9e-65520deab3a5",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T23:45:09.294956Z",
-     "iopub.status.busy": "2024-07-25T23:45:09.294483Z",
-     "iopub.status.idle": "2024-07-25T23:45:09.329935Z",
-     "shell.execute_reply": "2024-07-25T23:45:09.328680Z",
-     "shell.execute_reply.started": "2024-07-25T23:45:09.294917Z"
+     "iopub.execute_input": "2024-07-26T00:26:44.188306Z",
+     "iopub.status.busy": "2024-07-26T00:26:44.187916Z",
+     "iopub.status.idle": "2024-07-26T00:26:44.958847Z",
+     "shell.execute_reply": "2024-07-26T00:26:44.957986Z",
+     "shell.execute_reply.started": "2024-07-26T00:26:44.188266Z"
     }
    },
    "outputs": [],
@@ -146,15 +146,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 2,
    "id": "165d2698-ae4c-44f9-be75-24cf55028bde",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T23:37:25.378448Z",
-     "iopub.status.busy": "2024-07-25T23:37:25.378123Z",
-     "iopub.status.idle": "2024-07-25T23:37:25.399222Z",
-     "shell.execute_reply": "2024-07-25T23:37:25.398293Z",
-     "shell.execute_reply.started": "2024-07-25T23:37:25.378432Z"
+     "iopub.execute_input": "2024-07-26T00:26:46.734456Z",
+     "iopub.status.busy": "2024-07-26T00:26:46.733886Z",
+     "iopub.status.idle": "2024-07-26T00:26:46.753149Z",
+     "shell.execute_reply": "2024-07-26T00:26:46.752572Z",
+     "shell.execute_reply.started": "2024-07-26T00:26:46.734417Z"
     }
    },
    "outputs": [],
@@ -252,15 +252,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 3,
    "id": "65c18113-7067-41cf-81da-bb159f17e581",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T23:39:44.932720Z",
-     "iopub.status.busy": "2024-07-25T23:39:44.932431Z",
-     "iopub.status.idle": "2024-07-25T23:39:44.967543Z",
-     "shell.execute_reply": "2024-07-25T23:39:44.966823Z",
-     "shell.execute_reply.started": "2024-07-25T23:39:44.932699Z"
+     "iopub.execute_input": "2024-07-26T00:26:49.051643Z",
+     "iopub.status.busy": "2024-07-26T00:26:49.051331Z",
+     "iopub.status.idle": "2024-07-26T00:26:49.086011Z",
+     "shell.execute_reply": "2024-07-26T00:26:49.084808Z",
+     "shell.execute_reply.started": "2024-07-26T00:26:49.051617Z"
     }
    },
    "outputs": [
@@ -270,7 +270,7 @@
        "F1Dataset()"
       ]
      },
-     "execution_count": 9,
+     "execution_count": 3,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -282,15 +282,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 4,
    "id": "7d9de641-13ae-442e-bfa0-08d5efa5a1c1",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T23:40:17.495003Z",
-     "iopub.status.busy": "2024-07-25T23:40:17.494511Z",
-     "iopub.status.idle": "2024-07-25T23:40:17.528231Z",
-     "shell.execute_reply": "2024-07-25T23:40:17.527307Z",
-     "shell.execute_reply.started": "2024-07-25T23:40:17.494961Z"
+     "iopub.execute_input": "2024-07-26T00:27:05.636732Z",
+     "iopub.status.busy": "2024-07-26T00:27:05.636238Z",
+     "iopub.status.idle": "2024-07-26T00:27:05.670377Z",
+     "shell.execute_reply": "2024-07-26T00:27:05.669221Z",
+     "shell.execute_reply.started": "2024-07-26T00:27:05.636695Z"
     }
    },
    "outputs": [
@@ -300,27 +300,27 @@
        "DriverDNFTask(dataset=F1Dataset())"
       ]
      },
-     "execution_count": 10,
+     "execution_count": 4,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
-    "driver_dnf_task = DriverDNFTask(f1_dataset)\n",
+    "driver_dnf_task = DriverDNFTask(f1_dataset, cache_dir=\"./cache/driver_dnf\")\n",
     "driver_dnf_task"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 5,
    "id": "17dc774d-8aa7-4f2d-aad6-7c92a99f9209",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T23:40:37.796664Z",
-     "iopub.status.busy": "2024-07-25T23:40:37.796374Z",
-     "iopub.status.idle": "2024-07-25T23:40:38.178946Z",
-     "shell.execute_reply": "2024-07-25T23:40:38.178199Z",
-     "shell.execute_reply.started": "2024-07-25T23:40:37.796643Z"
+     "iopub.execute_input": "2024-07-26T00:27:11.575669Z",
+     "iopub.status.busy": "2024-07-26T00:27:11.575251Z",
+     "iopub.status.idle": "2024-07-26T00:27:11.929055Z",
+     "shell.execute_reply": "2024-07-26T00:27:11.928164Z",
+     "shell.execute_reply.started": "2024-07-26T00:27:11.575633Z"
     }
    },
    "outputs": [
@@ -331,10 +331,10 @@
       "Making task table for train split from scratch...\n",
       "(You can also use `get_task(..., download=True)` for tasks prepared by the RelBench team.)\n",
       "Loading Database object from /afs/cs.stanford.edu/u/ranjanr/.cache/relbench/rel-f1/db...\n",
-      "Done in 0.05 seconds.\n",
+      "Done in 0.04 seconds.\n",
       "Loading Database object from /afs/cs.stanford.edu/u/ranjanr/.cache/relbench/rel-f1/db...\n",
-      "Done in 0.03 seconds.\n",
-      "Done in 0.36 seconds.\n"
+      "Done in 0.02 seconds.\n",
+      "Done in 0.32 seconds.\n"
      ]
     },
     {
@@ -348,11 +348,11 @@
        "3     2004-07-05        47               1\n",
        "4     2004-06-05        31               0\n",
        "...          ...       ...             ...\n",
-       "11406 1998-02-07        14               0\n",
-       "11407 1984-08-19       159               0\n",
-       "11408 1982-06-01       207               0\n",
-       "11409 1965-07-30       422               0\n",
-       "11410 1955-04-24       652               0\n",
+       "11406 1977-04-28       228               0\n",
+       "11407 1977-06-27       259               0\n",
+       "11408 1957-05-13       611               1\n",
+       "11409 1955-05-24       611               0\n",
+       "11410 1954-06-28       683               0\n",
        "\n",
        "[11411 rows x 3 columns],\n",
        "  fkey_col_to_pkey_table={'driverId': 'drivers'},\n",
@@ -360,7 +360,7 @@
        "  time_col=date)"
       ]
      },
-     "execution_count": 11,
+     "execution_count": 5,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -387,15 +387,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": 6,
    "id": "942649f9-4527-4dba-817f-52eb44851189",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T23:43:16.952240Z",
-     "iopub.status.busy": "2024-07-25T23:43:16.951981Z",
-     "iopub.status.idle": "2024-07-25T23:43:17.251961Z",
-     "shell.execute_reply": "2024-07-25T23:43:17.250891Z",
-     "shell.execute_reply.started": "2024-07-25T23:43:16.952223Z"
+     "iopub.execute_input": "2024-07-26T00:27:15.890318Z",
+     "iopub.status.busy": "2024-07-26T00:27:15.890091Z",
+     "iopub.status.idle": "2024-07-26T00:27:16.157594Z",
+     "shell.execute_reply": "2024-07-26T00:27:16.156802Z",
+     "shell.execute_reply.started": "2024-07-26T00:27:15.890302Z"
     }
    },
    "outputs": [
@@ -422,7 +422,7 @@
        "  time_col=date)"
       ]
      },
-     "execution_count": 13,
+     "execution_count": 6,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -449,15 +449,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 16,
+   "execution_count": 7,
    "id": "3df1be1a-3024-4cec-9ef7-17b645af6aa5",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T23:45:21.801458Z",
-     "iopub.status.busy": "2024-07-25T23:45:21.801223Z",
-     "iopub.status.idle": "2024-07-25T23:45:21.818586Z",
-     "shell.execute_reply": "2024-07-25T23:45:21.818089Z",
-     "shell.execute_reply.started": "2024-07-25T23:45:21.801440Z"
+     "iopub.execute_input": "2024-07-26T00:27:21.252561Z",
+     "iopub.status.busy": "2024-07-26T00:27:21.252308Z",
+     "iopub.status.idle": "2024-07-26T00:27:21.269893Z",
+     "shell.execute_reply": "2024-07-26T00:27:21.269371Z",
+     "shell.execute_reply.started": "2024-07-26T00:27:21.252543Z"
     }
    },
    "outputs": [
@@ -467,7 +467,7 @@
        "['driver-position', 'driver-dnf', 'driver-top3', 'custom_driver-dnf']"
       ]
      },
-     "execution_count": 16,
+     "execution_count": 7,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -479,15 +479,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": 8,
    "id": "ca0fe893-b439-4f7f-aff1-f7e19200ecf9",
    "metadata": {
     "execution": {
-     "iopub.execute_input": "2024-07-25T23:45:59.551260Z",
-     "iopub.status.busy": "2024-07-25T23:45:59.550764Z",
-     "iopub.status.idle": "2024-07-25T23:45:59.584929Z",
-     "shell.execute_reply": "2024-07-25T23:45:59.583874Z",
-     "shell.execute_reply.started": "2024-07-25T23:45:59.551222Z"
+     "iopub.execute_input": "2024-07-26T00:27:23.056680Z",
+     "iopub.status.busy": "2024-07-26T00:27:23.056288Z",
+     "iopub.status.idle": "2024-07-26T00:27:23.086075Z",
+     "shell.execute_reply": "2024-07-26T00:27:23.085090Z",
+     "shell.execute_reply.started": "2024-07-26T00:27:23.056645Z"
     }
    },
    "outputs": [
@@ -497,7 +497,7 @@
        "DriverDNFTask(dataset=F1Dataset())"
       ]
      },
-     "execution_count": 17,
+     "execution_count": 8,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -537,6 +537,14 @@
    "source": [
     "Another form of customization is overriding the `_get_table` method directly. This can allow expressing tasks where the labels cannot be computed from the database alone, or splitting is not purely temporal."
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "71a366eb-0238-4919-a4ed-035d8e0a24c8",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {