From d5811590dc17395d62378f774e576aac7f0f0a95 Mon Sep 17 00:00:00 2001 From: Shubham Ugare Date: Wed, 23 Apr 2025 17:24:19 -0500 Subject: [PATCH] Print partial generation in dev mode before failing --- syncode/grammar_mask/grammar_constrainer.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/syncode/grammar_mask/grammar_constrainer.py b/syncode/grammar_mask/grammar_constrainer.py index 3fccce0e..ffcc95de 100644 --- a/syncode/grammar_mask/grammar_constrainer.py +++ b/syncode/grammar_mask/grammar_constrainer.py @@ -206,6 +206,9 @@ def _parse_partial_code(self, idx: int, partial_code: str, remainder_bytes: byte self._update_valid_state(partial_code, idx, res) except Exception as e: if self.dev_mode == True and accepted_generation: + logger.info("-"*50) + logger.info(f"Parsing failed!\nPartial code: {partial_output}\nParsed lexical tokens: {self.inc_parser.parsed_lexer_tokens}") + logger.info("-"*50) raise e elif self.parse_failed == False and accepted_generation: self.parse_failed = True