Limiting fraud dataset to speed up doc builds (#1646)

* fix fraud * update release notes * linting
alteryx · Jan 5, 2021 · d4739cf · d4739cf
1 parent 84b0214
commit d4739cf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 10 deletions.
diff --git a/docs/source/release_notes.rst b/docs/source/release_notes.rst
@@ -13,6 +13,7 @@ Release Notes
         * Addressed stacked ensemble component for ``scikit-learn`` v0.24 support by setting ``shuffle=True`` for default CV :pr:`1613`
         * Fix bug where ``Imputer`` reset the index on ``X`` :pr:`1590`
         * Fixed AutoMLSearch stacktrace when a cutom objective was passed in as a primary objective or additional objective :pr:`1575`
+        * Limit ``load_fraud`` dataset loaded into ``automl.ipynb`` :pr:`1646`
         * Fix custom index bug for ``MAPE`` objective :pr:`1641`
         * Fixed bug where time series baseline estimators were not receiving ``gap`` and ``max_delay`` in ``AutoMLSearch`` :pr:`1645`
     * Changes

diff --git a/docs/source/user_guide/automl.ipynb b/docs/source/user_guide/automl.ipynb
@@ -58,7 +58,7 @@
    "source": [
     "import evalml\n",
     "from evalml.utils import infer_feature_types\n",
-    "X, y = evalml.demos.load_fraud(return_pandas=True)\n",
+    "X, y = evalml.demos.load_fraud(n_rows=1000, return_pandas=True)\n",
     "X = infer_feature_types(X, feature_types={'provider': 'categorical'})"
    ]
   },
@@ -211,9 +211,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "scrolled": false
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "## start_iteration_callback example function\n",
@@ -232,9 +230,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "scrolled": false
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "## add_result_callback example function\n",
@@ -425,9 +421,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "scrolled": false
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "X, y = evalml.demos.load_breast_cancer()\n",