chore: fix flaky HyperOpt NB (#2198)

microsoft · Apr 10, 2024 · d30d57b · d30d57b
1 parent 8e4f326
commit d30d57b
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/docs/Explore Algorithms/Hyperparameter Tuning/HyperOpt.ipynb b/docs/Explore Algorithms/Hyperparameter Tuning/HyperOpt.ipynb
@@ -148,7 +148,8 @@
    "source": [
     "import numpy as np\n",
     "import pandas as pd\n",
-    "from sklearn.datasets import fetch_california_housing"
+    "from sklearn.datasets import fetch_california_housing\n",
+    "import time"
    ]
   },
   {
@@ -157,7 +158,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "california = fetch_california_housing()\n",
+    "try:\n",
+    "    california = fetch_california_housing()\n",
+    "except EOFError:\n",
+    "    print(\"Encountered EOFError while downloading, retrying once...\")\n",
+    "    time.sleep(5)\n",
+    "    california = fetch_california_housing()\n",
     "\n",
     "feature_cols = [\"f\" + str(i) for i in range(california.data.shape[1])]\n",
     "header = [\"target\"] + feature_cols\n",