From e42cedd7dc92004462b422803ea011b99f1d7ebc Mon Sep 17 00:00:00 2001
From: Lynn <firemagelynn@gmail.com>
Date: Sun, 22 Oct 2023 21:31:20 -0700
Subject: [PATCH] mypy

---
 src/python/tokenizer_script.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/python/tokenizer_script.py b/src/python/tokenizer_script.py
index 5de70c7..62e55a6 100644
--- a/src/python/tokenizer_script.py
+++ b/src/python/tokenizer_script.py
@@ -11,7 +11,7 @@
 import typing
 
 
-DEBUG = bool(int(os.getenv("DEBUG")))
+DEBUG = bool(int(os.getenv("DEBUG", "0")))
 
 
 @dataclasses.dataclass(frozen=True)
@@ -59,12 +59,12 @@ def tokenize_sql(self, sql: list[str]) -> list[SQLTokens]:
                 raise ValueError(f"Unknown worker function: {this_worker_str}")
 
         # tokenize args
-        args_list = []
+        args_list: list[list] = []
         for i, sentence in enumerate(args_strs):
             args_list.append([])
-            word_start = 0
+            word_start: int | None = 0
             inside_list = False
-            string_start = (None, None)
+            string_start: tuple[int | None, str | None] = (None, None)
             for k, letter in enumerate(sentence):
                 if (string_start[0] is None) and (letter in ["'", '"']):
                     if DEBUG: