From 70e8357d16d52b8dcfe2b3cd5525dd1015e4b6ef Mon Sep 17 00:00:00 2001
From: okanyenigun <okanyenigun@gmail.com>
Date: Sun, 19 Jan 2025 23:12:40 +0300
Subject: [PATCH] Enhance compatibility with Gemma 2 LLMs by refining code
 preprocessing in executor.py

---
 lida/components/executor.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lida/components/executor.py b/lida/components/executor.py
index 7be321096..35bed2b57 100644
--- a/lida/components/executor.py
+++ b/lida/components/executor.py
@@ -20,6 +20,8 @@ def preprocess_code(code: str) -> str:
     code = code.replace("<imports>", "")
     code = code.replace("<stub>", "")
     code = code.replace("<transforms>", "")
+    code = code.replace(" import", "import")
+    code = code.replace(" def", "def")
 
     # remove all text after chart = plot(data)
     if "chart = plot(data)" in code: