Update oai_completion.ipynb (#623)

Missing import
2025-12-25 05:59:19 +00:00 · 2023-11-10 16:50:59 -08:00 · 2023-11-10 16:50:59 -08:00 · b1e69bf2f2
commit b1e69bf2f2
parent d5cbc2b3ba
1 changed files with 2 additions and 0 deletions
--- a/notebook/oai_completion.ipynb
+++ b/notebook/oai_completion.ipynb
@ -404,6 +404,8 @@
    "Users can specify tuning data, optimization metric, optimization mode, evaluation function, search spaces, etc. The default search space is:\n",
    "\n",
    "```python\n",
+    "from flaml import tune\n",
+    "\n",
    "default_search_space = {\n",
    "    \"model\": tune.choice([\n",
    "        \"text-ada-001\",\n",