From 7ef466e88d95097c680bc66bb4fc6fb610ae4a44 Mon Sep 17 00:00:00 2001
From: Jonathan Slenders <jonathan@slenders.be>
Date: Thu, 9 Nov 2023 10:40:24 +0100
Subject: [PATCH] Add types for `pygments.lex`. (#10998)

---
 stubs/Pygments/pygments/__init__.pyi | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/stubs/Pygments/pygments/__init__.pyi b/stubs/Pygments/pygments/__init__.pyi
index ce36f3c07..03d982778 100644
--- a/stubs/Pygments/pygments/__init__.pyi
+++ b/stubs/Pygments/pygments/__init__.pyi
@@ -1,14 +1,17 @@
 from _typeshed import SupportsWrite
+from collections.abc import Iterator
 from typing import TypeVar, overload
 
 from pygments.formatter import Formatter
+from pygments.lexer import Lexer
+from pygments.token import _TokenType
 
 _T = TypeVar("_T", str, bytes)
 
 __version__: str
 __all__ = ["lex", "format", "highlight"]
 
-def lex(code, lexer): ...
+def lex(code: str, lexer: Lexer) -> Iterator[tuple[_TokenType, str]]: ...
 @overload
 def format(tokens, formatter: Formatter[_T], outfile: SupportsWrite[_T]) -> None: ...
 @overload