diff --git a/docs/Notebooks/TSEvo_tensorflow.ipynb b/docs/Notebooks/TSEvo_tensorflow.ipynb
index 5dcd8f4..7c93995 100644
--- a/docs/Notebooks/TSEvo_tensorflow.ipynb
+++ b/docs/Notebooks/TSEvo_tensorflow.ipynb
@@ -6,6 +6,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "!pip install https://github.com/gkhayes/mlrose/archive/refs/heads/master.zip\n",
     "import pandas as pd\n",
     "import numpy as np\n",
     "import torch\n",
diff --git a/docs/Notebooks/TSEvo_torch_LSTM.ipynb b/docs/Notebooks/TSEvo_torch_LSTM.ipynb
index 060e355..92813ad 100644
--- a/docs/Notebooks/TSEvo_torch_LSTM.ipynb
+++ b/docs/Notebooks/TSEvo_torch_LSTM.ipynb
@@ -19,6 +19,15 @@
     "[2] Wachter, Sandra, Brent Mittelstadt, and Chris Russell. \"Counterfactual explanations without opening the black box: Automated decisions and the GDPR.\" Harv. JL & Tech. 31 (2017): 841."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install https://github.com/gkhayes/mlrose/archive/refs/heads/master.zip"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,
diff --git a/docs/Notebooks/TSEvo_torch_Multivariate.ipynb b/docs/Notebooks/TSEvo_torch_Multivariate.ipynb
index ae8584e..d40488b 100644
--- a/docs/Notebooks/TSEvo_torch_Multivariate.ipynb
+++ b/docs/Notebooks/TSEvo_torch_Multivariate.ipynb
@@ -19,6 +19,15 @@
     "[2] Wachter, Sandra, Brent Mittelstadt, and Chris Russell. \"Counterfactual explanations without opening the black box: Automated decisions and the GDPR.\" Harv. JL & Tech. 31 (2017): 841."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install https://github.com/gkhayes/mlrose/archive/refs/heads/master.zip"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,
diff --git a/setup.py b/setup.py
index 7d07874..f3253fa 100644
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,7 @@
 
 # Package requirements.
 base_packages = [
-    "mlrose @ https://github.com/gkhayes/mlrose/archive/refs/heads/master.zip ",
+    #"mlrose @ https://github.com/gkhayes/mlrose/archive/refs/heads/master.zip",
     #"mlrose>=1.3.0,< 2.0",
     "scikit-learn==1.3.0",
     # "scikit-surprise==1.1.1",
@@ -64,7 +64,8 @@
     "pre-commit>=2.9.2,< 3.0",
     "pytest>=4.5.0,< 5.0",
     "pytest-cov>=2.6.1,< 3.0",
-    "pyupgrade>=3.2.0,< 4.0"
+    "pyupgrade>=3.2.0,< 4.0",
+    "mlrose @ https://github.com/gkhayes/mlrose/archive/refs/heads/master.zip"
 ]
 
 docs_packages = [