Help the type checker a bit with recursive type definitions

This commit is contained in:
Dave Halter
2026-02-04 15:58:38 +01:00
parent 6fbeec9e2f
commit 07fb4584a3

View File

@@ -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):