Merge pull request #87 from automl/fix_doc_example

Fix examples for docs
automl · May 14, 2024 · 3b3a53a · 3b3a53a
2 parents f6f1992 + 74fbf09
commit 3b3a53a
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/docs/getting_started/single_worker.md b/docs/getting_started/single_worker.md
@@ -44,7 +44,7 @@ optimizer = DEHB(
 
 # Run optimization for 1 bracket. Output files will be saved to ./logs
 traj, runtime, history = optimizer.run(brackets=1, verbose=True)
-config, fitness, runtime, fidelity, _ = history[0]
+config_id, config, fitness, runtime, fidelity, _ = history[0]
 print("config", config)
 print("fitness", fitness)
 print("runtime", runtime)

diff --git a/examples/01.1_Optimizing_RandomForest_using_DEHB.ipynb b/examples/01.1_Optimizing_RandomForest_using_DEHB.ipynb
@@ -395,7 +395,7 @@
     "\n",
     "# Last recorded function evaluation\n",
     "last_eval = history[-1]\n",
-    "config, score, cost, fidelity, _info = last_eval\n",
+    "config_id, config, score, cost, fidelity, _info = last_eval\n",
     "\n",
     "print(\"Last evaluated configuration, \")\n",
     "print(dehb.vector_to_configspace(config), end=\"\")\n",

diff --git a/examples/01.2_Optimizing_RandomForest_using_Ask_Tell.ipynb b/examples/01.2_Optimizing_RandomForest_using_Ask_Tell.ipynb
@@ -418,7 +418,7 @@
     "\n",
     "# Last recorded function evaluation\n",
     "last_eval = history[-1]\n",
-    "config, score, cost, fidelity, _info = last_eval\n",
+    "config_id, config, score, cost, fidelity, _info = last_eval\n",
     "\n",
     "print(\"Last evaluated configuration, \")\n",
     "print(dehb.vector_to_configspace(config), end=\"\")\n",
@@ -634,7 +634,7 @@
     "\n",
     "# Last recorded function evaluation\n",
     "last_eval = history[-1]\n",
-    "config, score, cost, fidelity, _info = last_eval\n",
+    "config_id, config, score, cost, fidelity, _info = last_eval\n",
     "\n",
     "print(\"Last evaluated configuration, \")\n",
     "print(dehb.vector_to_configspace(config), end=\"\")\n",

diff --git a/pyproject.toml b/pyproject.toml
@@ -54,7 +54,8 @@ dev = [
   "pre-commit",
 ]
 examples = [
-  "scikit-learn"
+  "scikit-learn",
+  "matplotlib",
 ]
 
 [tool.pytest.ini_options]