From b8f54d79e32d1ea0c0606205279b163ca738832e Mon Sep 17 00:00:00 2001
From: Hayden <64056131+hay-kot@users.noreply.github.com>
Date: Fri, 4 Jul 2025 11:55:55 -0500
Subject: [PATCH] fix type annotation

---
 dev/code-generation/utils/template.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/dev/code-generation/utils/template.py b/dev/code-generation/utils/template.py
index 6312426e2..a2dd28cb7 100644
--- a/dev/code-generation/utils/template.py
+++ b/dev/code-generation/utils/template.py
@@ -35,7 +35,7 @@ class CodeSlicer:
     start: int
     end: int
 
-    indentation: str
+    indentation: str | None
     text: list[str]
 
     _next_line = None
@@ -47,7 +47,7 @@ class CodeSlicer:
 
     def push_line(self, string: str) -> None:
         self._next_line = self._next_line or self.start + 1
-        self.text.insert(self._next_line, self.indentation + string + "\n")
+        self.text.insert(self._next_line, self.indentation or "" + string + "\n")
         self._next_line += 1
 
 
@@ -55,7 +55,7 @@ def get_indentation_of_string(line: str, comment_char: str = "//|#") -> str:
     return re.sub(rf"{comment_char}.*", "", line).removesuffix("\n")
 
 
-def find_start_end(file_text: list[str], gen_id: str) -> tuple[int, int, str]:
+def find_start_end(file_text: list[str], gen_id: str) -> tuple[int, int, str | None]:
     start = None
     end = None
     indentation = None