chore: correct wrong rebase (#20)

Corrects some changes that were lost on a previous rebase.
TheVaultdweller13 · Jan 2, 2025 · 3a297b0 · 3a297b0
1 parent 10b8c9d
commit 3a297b0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/src/parser.py b/src/parser.py
@@ -12,9 +12,7 @@ def __init__(
         self,
         grammar=grammar.imperivm,
         visitor=visitor.ImperivmVisitor(),
-        preprocessor=preprocessor.PipelinePreprocessor(
-            [preprocessor.CommentsPreprocessor()]
-        ),
+        preprocessor=preprocessor.PipelinePreprocessor([preprocessor.CommentsPreprocessor()]),
     ):
         self.grammar = grammar
         self.visitor = visitor

diff --git a/src/preprocessor.py b/src/preprocessor.py
@@ -5,16 +5,14 @@
 
 class Preprocessor(ABC):
     @abstractmethod
-    def process(self, program: str):
+    def process(self, program: str) -> str:
         pass
 
 
 class CommentsPreprocessor(Preprocessor):
     @override
     def process(self, program: str):
-        cleaned_program = map(
-            lambda line: re.sub(r"#.*$", "", line), program.splitlines()
-        )
+        cleaned_program = map(lambda line: re.sub(r"#.*$", "", line), program.splitlines())
         return "\n".join(cleaned_program)