fix tests

alteryx · Aug 7, 2023 · 759a2eb · 759a2eb
1 parent c2c16a9
commit 759a2eb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/docs/source/examples/predict_next_purchase.ipynb b/docs/source/examples/predict_next_purchase.ipynb
@@ -26,6 +26,7 @@
     "from demo.next_purchase import load_sample\n",
     "from matplotlib.pyplot import subplots\n",
     "import composeml as cp\n",
+    "import pandas as pd\n",
     "import featuretools as ft\n",
     "import evalml"
    ]
@@ -129,7 +130,7 @@
     "    df.sort_values(\"order_time\"),\n",
     "    num_examples_per_instance=-1,\n",
     "    product_name=\"Banana\",\n",
-    "    minimum_data=\"3d\",\n",
+    "    minimum_data=pd.Timedelta(\"3d\"),\n",
     "    verbose=False,\n",
     ")\n",
     "\n",

diff --git a/docs/source/user_guide/using_label_transforms.ipynb b/docs/source/user_guide/using_label_transforms.ipynb
@@ -28,6 +28,7 @@
    "outputs": [],
    "source": [
     "import composeml as cp\n",
+    "import pandas as pd\n",
     "\n",
     "\n",
     "def total_spent(df):\n",
@@ -44,7 +45,7 @@
     "labels = label_maker.search(\n",
     "    cp.demos.load_transactions(),\n",
     "    num_examples_per_instance=10,\n",
-    "    minimum_data=\"2h\",\n",
+    "    minimum_data=pd.Timedelta(\"2h\"),\n",
     "    gap=\"2min\",\n",
     "    verbose=True,\n",
     ")"