[tune] fix hebo_example. (ray-project#26439)

Fixes a bug in the the ipython notebook.
pabloem · Jul 11, 2022 · c97d65e · c97d65e
1 parent b01e11d
commit c97d65e
Show file tree

Hide file tree

Showing 2 changed files with 56 additions and 14 deletions.
diff --git a/doc/source/tune/examples/BUILD b/doc/source/tune/examples/BUILD
@@ -22,7 +22,6 @@ py_test_run_all_notebooks(
         "nyc_taxi_basic_processing.ipynb", # REGRESSION
         "ocr_example.ipynb", # REGRESSION
         "bohb_example.ipynb", # REGRESSION
-        "hebo_example.ipynb", # REGRESSION
         "horovod_simple.ipynb", # REGRESSION
         "hyperopt_example.ipynb", # REGRESSION
         "sigopt_example.ipynb", # REGRESSION

diff --git a/doc/source/tune/examples/hebo_example.ipynb b/doc/source/tune/examples/hebo_example.ipynb
@@ -21,7 +21,10 @@
    "metadata": {
     "tags": [
      "remove-cell"
-    ]
+    ],
+    "vscode": {
+     "languageId": "python"
+    }
    },
    "outputs": [],
    "source": [
@@ -46,7 +49,10 @@
    "metadata": {
     "tags": [
      "hide-input"
-    ]
+    ],
+    "vscode": {
+     "languageId": "python"
+    }
    },
    "outputs": [],
    "source": [
@@ -73,7 +79,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "d654948a",
-   "metadata": {},
+   "metadata": {
+    "vscode": {
+     "languageId": "python"
+    }
+   },
    "outputs": [],
    "source": [
     "def evaluate(step, width, height, activation):\n",
@@ -95,7 +105,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "b7e5f38b",
-   "metadata": {},
+   "metadata": {
+    "vscode": {
+     "languageId": "python"
+    }
+   },
    "outputs": [],
    "source": [
     "def objective(config):\n",
@@ -111,7 +125,10 @@
    "metadata": {
     "tags": [
      "remove-cell"
-    ]
+    ],
+    "vscode": {
+     "languageId": "python"
+    }
    },
    "outputs": [],
    "source": [
@@ -133,7 +150,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "9aaf222e",
-   "metadata": {},
+   "metadata": {
+    "vscode": {
+     "languageId": "python"
+    }
+   },
    "outputs": [],
    "source": [
     "\n",
@@ -169,7 +190,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "ea2a405c",
-   "metadata": {},
+   "metadata": {
+    "vscode": {
+     "languageId": "python"
+    }
+   },
    "outputs": [],
    "source": [
     "num_samples = 1000"
@@ -182,7 +207,10 @@
    "metadata": {
     "tags": [
      "remove-cell"
-    ]
+    ],
+    "vscode": {
+     "languageId": "python"
+    }
    },
    "outputs": [],
    "source": [
@@ -203,14 +231,18 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "0a530e21",
-   "metadata": {},
+   "metadata": {
+    "vscode": {
+     "languageId": "python"
+    }
+   },
    "outputs": [],
    "source": [
     "search_config = {\n",
     "    \"steps\": 100,\n",
     "    \"width\": tune.uniform(0, 20),\n",
     "    \"height\": tune.uniform(-100, 100),\n",
-    "    \"activation\": tune.choice([\"relu, tanh\"])\n",
+    "    \"activation\": tune.choice([\"relu\", \"tanh\"])\n",
     "}"
    ]
   },
@@ -226,7 +258,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "3fb63ef3",
-   "metadata": {},
+   "metadata": {
+    "vscode": {
+     "languageId": "python"
+    }
+   },
    "outputs": [],
    "source": [
     "analysis = tune.run(\n",
@@ -252,7 +288,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "8b8a23e4",
-   "metadata": {},
+   "metadata": {
+    "vscode": {
+     "languageId": "python"
+    }
+   },
    "outputs": [],
    "source": [
     "print(\"Best hyperparameters found were: \", analysis.best_config)"
@@ -265,7 +305,10 @@
    "metadata": {
     "tags": [
      "remove-cell"
-    ]
+    ],
+    "vscode": {
+     "languageId": "python"
+    }
    },
    "outputs": [],
    "source": [