diff --git a/stdlib/2and3/cProfile.pyi b/stdlib/2and3/cProfile.pyi index b31ea0337..dea70bf48 100644 --- a/stdlib/2and3/cProfile.pyi +++ b/stdlib/2and3/cProfile.pyi @@ -1,7 +1,7 @@ -from typing import Any, Callable, Dict, Optional, TypeVar +from typing import Any, Callable, Dict, Optional, TypeVar, Union -def run(statement: str, filename: Optional[str] = ..., sort: int = ...) -> None: ... -def runctx(statement: str, globals: Dict[str, Any], locals: Dict[str, Any], filename: Optional[str] = ..., sort: int = ...) -> None: ... +def run(statement: str, filename: Optional[str] = ..., sort: Union[str, int] = ...) -> None: ... +def runctx(statement: str, globals: Dict[str, Any], locals: Dict[str, Any], filename: Optional[str] = ..., sort: Union[str, int] = ...) -> None: ... _SelfT = TypeVar('_SelfT', bound='Profile') _T = TypeVar('_T') @@ -10,7 +10,7 @@ class Profile: def __init__(self, custom_timer: Callable[[], float] = ..., time_unit: float = ..., subcalls: bool = ..., builtins: bool = ...) -> None: ... def enable(self) -> None: ... def disable(self) -> None: ... - def print_stats(self, sort: int = ...) -> None: ... + def print_stats(self, sort: Union[str, int] = ...) -> None: ... def dump_stats(self, file: str) -> None: ... def create_stats(self) -> None: ... def run(self: _SelfT, cmd: str) -> _SelfT: ... diff --git a/stdlib/2and3/profile.pyi b/stdlib/2and3/profile.pyi index 21a7c8fd4..72952a021 100644 --- a/stdlib/2and3/profile.pyi +++ b/stdlib/2and3/profile.pyi @@ -1,7 +1,7 @@ -from typing import Any, Callable, Dict, Optional, TypeVar +from typing import Any, Callable, Dict, Optional, TypeVar, Union -def run(statement: str, filename: Optional[str] = ..., sort: int = ...) -> None: ... -def runctx(statement: str, globals: Dict[str, Any], locals: Dict[str, Any], filename: Optional[str] = ..., sort: int = ...) -> None: ... +def run(statement: str, filename: Optional[str] = ..., sort: Union[str, int] = ...) -> None: ... +def runctx(statement: str, globals: Dict[str, Any], locals: Dict[str, Any], filename: Optional[str] = ..., sort: Union[str, int] = ...) -> None: ... _SelfT = TypeVar('_SelfT', bound='Profile') _T = TypeVar('_T') @@ -11,7 +11,7 @@ class Profile: def set_cmd(self, cmd: str) -> None: ... def simulate_call(self, name: str) -> None: ... def simulate_cmd_complete(self) -> None: ... - def print_stats(self, sort: int = ...) -> None: ... + def print_stats(self, sort: Union[str, int] = ...) -> None: ... def dump_stats(self, file: str) -> None: ... def create_stats(self) -> None: ... def snapshot_stats(self) -> None: ...