From 1c4964700274ae0d8667456d4a5df29c63238054 Mon Sep 17 00:00:00 2001
From: xcharleslin <4212216+xcharleslin@users.noreply.github.com>
Date: Wed, 21 Jun 2023 15:36:48 -0700
Subject: [PATCH] In CI, limit tutorial to 500 rows (#1076)

Co-authored-by: Xiayue Charles Lin <charles@eventualcomputing.com>
---
 ...ft_tutorial_embeddings_stackexchange.ipynb | 26 ++++++++++++-------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/tutorials/embeddings/daft_tutorial_embeddings_stackexchange.ipynb b/tutorials/embeddings/daft_tutorial_embeddings_stackexchange.ipynb
index 2ffc9a5b72..f41c1de165 100644
--- 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": {