In CI, limit tutorial to 500 rows (#1076)

Co-authored-by: Xiayue Charles Lin <[email protected]>
Eventual-Inc · Jun 21, 2023 · 1c49647 · 1c49647
1 parent 2bcaae4
commit 1c49647
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/tutorials/embeddings/daft_tutorial_embeddings_stackexchange.ipynb b/tutorials/embeddings/daft_tutorial_embeddings_stackexchange.ipynb
@@ -43,7 +43,20 @@
    ]
   },
   {
-   "attachments": {},
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3f40dafa",
+   "metadata": {
+    "tags": [
+     "parameters"
+    ]
+   },
+   "outputs": [],
+   "source": [
+    "CI = False"
+   ]
+  },
+  {
    "cell_type": "markdown",
    "id": "2a45a28c",
    "metadata": {},
@@ -93,6 +106,9 @@
     "\n",
     "df = daft.read_json(SAMPLE_DATA_PATH)\n",
     "\n",
+    "if CI:\n",
+    "    df = df.limit(500)\n",
+    "\n",
     "print(df)"
    ]
   },
@@ -433,14 +449,6 @@
     "\n",
     "# You can get daft at [getdaft.io](https://getdaft.io)."
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "3d1913ae",
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {