From 07fb4584a3c5cf31a78c5cdb52ac19b8f9843a99 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Wed, 4 Feb 2026 15:58:38 +0100 Subject: [PATCH] Help the type checker a bit with recursive type definitions --- parso/grammar.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/parso/grammar.py b/parso/grammar.py index af4e3b6..0a48795 100644 --- a/parso/grammar.py +++ b/parso/grammar.py @@ -1,6 +1,6 @@ import hashlib import os -from typing import Generic, TypeVar, Union, Dict, Optional, Any +from typing import Generic, TypeVar, Union, Dict, Optional, Any, Iterator from pathlib import Path from parso._compatibility import is_pypy @@ -8,7 +8,7 @@ from parso.pgen2 import generate_grammar from parso.utils import split_lines, python_bytes_to_unicode, \ PythonVersionInfo, parse_version_string from parso.python.diff import DiffParser -from parso.python.tokenize import tokenize_lines, tokenize +from parso.python.tokenize import tokenize_lines, tokenize, PythonToken from parso.python.token import PythonTokenTypes from parso.cache import parser_cache, load_module, try_to_save_module from parso.parser import BaseParser @@ -223,7 +223,7 @@ class PythonGrammar(Grammar): ) self.version_info = version_info - def _tokenize_lines(self, lines, **kwargs): + def _tokenize_lines(self, lines, **kwargs) -> Iterator[PythonToken]: return tokenize_lines(lines, version_info=self.version_info, **kwargs) def _tokenize(self, code):