#1017

mabel-dev · Jun 27, 2023 · 30fced2 · 30fced2
1 parent c88bc07
commit 30fced2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 18 deletions.
diff --git a/opteryx/third_party/README.md b/opteryx/third_party/README.md
diff --git a/opteryx/third_party/mbleven.py b/opteryx/third_party/mbleven.py
@@ -74,10 +74,9 @@ def compare(str1: str, str2: str) -> int:
 def check_model(str1: str, str2: str, len1: int, len2: int, model) -> int:
     """Check if the model can transform str1 into str2"""
 
-    idx1, idx2 = 0, 0
-    cost, pad = 0, 0
+    cost, idx1, idx2 = 0, 0, 0
     while idx1 < len1 and idx2 < len2:
-        if str1[idx1] != str2[idx2 - pad]:
+        if str1[idx1] != str2[idx2]:
             cost += 1
             if cost > 2:
                 return cost
@@ -90,10 +89,8 @@ def check_model(str1: str, str2: str, len1: int, len2: int, model) -> int:
             elif option == REPLACE:
                 idx1 += 1
                 idx2 += 1
-                pad = 0
         else:
             idx1 += 1
             idx2 += 1
-            pad = 0
 
     return cost + (len1 - idx1) + (len2 - idx2)
diff --git a/opteryx/utils/formatter.py b/opteryx/utils/formatter.py
@@ -125,8 +125,4 @@ def _replacer(match):
     formatted_sql = formatted_sql.replace(" \033[38;5;102m,", "\033[38;5;102m,")
     formatted_sql = formatted_sql.replace(" \033[38;5;102m;", "\033[38;5;102m;")
 
-    return (
-        color_comments(formatted_sql)
-        .strip()
-        .replace("\n", " \033[7;38;2;98;114;164m\033[3m↵\033[0m ")
-    )
+    return color_comments(formatted_sql).strip()