mypy fix
Signed-off-by: Gregory Becker <becker33@llnl.gov>
This commit is contained in:
parent
452a84d9eb
commit
231d537a2e
@ -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()
|
||||||
|
Loading…
Reference in New Issue
Block a user