From 625c46e5cc34b4ba97598220f9fa6888b9d2e832 Mon Sep 17 00:00:00 2001
From: Andreas Volkmann <avolkmann@icloud.com>
Date: Fri, 10 Nov 2023 14:33:29 -0800
Subject: [PATCH] Update oai_completion.ipynb

Missing import
---
 notebook/oai_completion.ipynb | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/notebook/oai_completion.ipynb b/notebook/oai_completion.ipynb
index f25db693fb81..d3aaacb17cb8 100644
--- 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",