From b0e1cef4dd9651f6470174f5d90ea9f18b5b8b97 Mon Sep 17 00:00:00 2001 From: Alexandre Viau Date: Wed, 13 Oct 2021 16:11:44 -0400 Subject: [PATCH] freezegun: _freeze_time.__enter__ return a freeze factory (#6164) --- stubs/freezegun/freezegun/api.pyi | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stubs/freezegun/freezegun/api.pyi b/stubs/freezegun/freezegun/api.pyi index df10e569a..d871d9258 100644 --- a/stubs/freezegun/freezegun/api.pyi +++ b/stubs/freezegun/freezegun/api.pyi @@ -39,7 +39,7 @@ class _freeze_time: def __call__(self, func: Callable[..., Awaitable[_T]]) -> Callable[..., Awaitable[_T]]: ... @overload def __call__(self, func: Callable[..., _T]) -> Callable[..., _T]: ... - def __enter__(self) -> Any: ... + def __enter__(self) -> FrozenDateTimeFactory | StepTickTimeFactory: ... def __exit__(self, *args: Any) -> None: ... def start(self) -> Any: ... def stop(self) -> None: ...