1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
"""
Python polyfills for builtins
"""
from __future__ import annotations
import builtins
import functools
import operator
from typing import Iterable, TypeVar
from ..decorators import substitute_in_graph
__all__ = [
"all",
"any",
"enumerate",
"sum",
]
_T = TypeVar("_T")
@substitute_in_graph(builtins.all, can_constant_fold_through=True)
def all(iterable: Iterable[object], /) -> bool:
for elem in iterable:
if not elem:
return False
return True
@substitute_in_graph(builtins.any, can_constant_fold_through=True)
def any(iterable: Iterable[object], /) -> bool:
for elem in iterable:
if elem:
return True
return False
@substitute_in_graph(builtins.enumerate, is_embedded_type=True) # type: ignore[arg-type]
def enumerate(iterable: Iterable[_T], start: int = 0) -> Iterable[tuple[int, _T]]:
if not isinstance(start, int):
raise TypeError(
f"{type(start).__name__!r} object cannot be interpreted as an integer"
)
for x in iterable:
yield start, x
start += 1
@substitute_in_graph(builtins.sum, can_constant_fold_through=True) # type: ignore[arg-type]
def sum(iterable: Iterable[_T], /, start: _T = 0) -> _T: # type: ignore[assignment]
return functools.reduce(operator.add, iterable, start)
|
