From a273b933242abf567ffcbec770094833dc75754c Mon Sep 17 00:00:00 2001
From: Tuana Celik <tuana.celik@deepset.ai>
Date: Fri, 27 Sep 2024 17:04:46 +0200
Subject: [PATCH] moving import up

---
 notebooks/query_decomposition.ipynb | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/notebooks/query_decomposition.ipynb b/notebooks/query_decomposition.ipynb
index 5fd1223..45a768d 100644
--- a/notebooks/query_decomposition.ipynb
+++ b/notebooks/query_decomposition.ipynb
@@ -53,7 +53,8 @@
         "from datasets import load_dataset\n",
         "from openai import OpenAI, Stream\n",
         "from openai.types.chat import ChatCompletion, ChatCompletionChunk\n",
-        "from typing import List, Any, Dict, Optional, Callable, Union"
+        "from typing import List, Any, Dict, Optional, Callable, Union\n",
+        "from pydantic import BaseModel"
       ]
     },
     {
@@ -223,8 +224,6 @@
       },
       "outputs": [],
       "source": [
-        "from pydantic import BaseModel\n",
-        "\n",
         "class Question(BaseModel):\n",
         "    question: str\n",
         "    answer: Optional[str] = None\n",