Signed-off-by: Gregory Becker <becker33@llnl.gov>
This commit is contained in:
Gregory Becker 2025-05-15 14:53:35 -07:00
parent 452a84d9eb
commit 231d537a2e
No known key found for this signature in database
GPG Key ID: 2362541F6D14ED84

View File

@ -63,7 +63,7 @@
import sys import sys
import traceback import traceback
import warnings import warnings
from typing import Iterator, List, Optional, Tuple, Union from typing import Iterable, Iterator, List, Optional, Tuple, Union
from llnl.util.tty import color from llnl.util.tty import color
@ -202,7 +202,10 @@ def expect(self, *kinds: SpecTokens):
def push(self, token_stream: Iterator[Token]): def push(self, token_stream: Iterator[Token]):
# New tokens need to go before next_token, which comes before the rest of the stream # New tokens need to go before next_token, which comes before the rest of the stream
self.token_stream = itertools.chain(token_stream, (self.next_token,), self.token_stream) next_token_iterator: Iterable[Token] = (
iter((self.next_token,)) if self.next_token else iter(())
)
self.token_stream = itertools.chain(token_stream, next_token_iterator, self.token_stream)
self.current_token = None self.current_token = None
self.next_token = None self.next_token = None
self.advance() self.advance()