diff --git a/cldk/utils/sanitization/java/treesitter_sanitizer.py b/cldk/utils/sanitization/java/treesitter_sanitizer.py index 4756025..e87c88c 100644 --- a/cldk/utils/sanitization/java/treesitter_sanitizer.py +++ b/cldk/utils/sanitization/java/treesitter_sanitizer.py @@ -120,7 +120,10 @@ def remove_unused_imports(self, sanitized_code: str) -> str: import_statement: Captures = self.__javasitter.frame_query_and_capture_output( query="((scoped_identifier) @scoped_identifier)", code_to_process=import_declaration.node.text.decode() ) - import_str = import_statement.captures[0].node.text.decode() + try: + import_str = import_statement.captures[0].node.text.decode() + except IndexError: + continue if import_str.split(".")[-1] not in ids_and_typeids: unused_imports.add(import_declaration.node.text.decode())