From 231d537a2efb8e86c4d9e841a5331d3d1aa6506d Mon Sep 17 00:00:00 2001 From: Gregory Becker Date: Thu, 15 May 2025 14:53:35 -0700 Subject: [PATCH] mypy fix Signed-off-by: Gregory Becker --- lib/spack/spack/spec_parser.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/lib/spack/spack/spec_parser.py b/lib/spack/spack/spec_parser.py index f619a93847d..fb9a1afb1cc 100644 --- a/lib/spack/spack/spec_parser.py +++ b/lib/spack/spack/spec_parser.py @@ -63,7 +63,7 @@ import sys import traceback 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 @@ -202,7 +202,10 @@ def expect(self, *kinds: SpecTokens): def push(self, token_stream: Iterator[Token]): # 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.next_token = None self.advance()