diff --git a/stubs/sklearn/_config.pyi b/stubs/sklearn/_config.pyi
index ab99fa7d..2b0a128c 100644
--- a/stubs/sklearn/_config.pyi
+++ b/stubs/sklearn/_config.pyi
@@ -1,4 +1,5 @@
-from collections.abc import Iterator
+from collections.abc import Generator
+from contextlib import contextmanager
 from typing import Literal
 
 from ._typing import Int
@@ -19,6 +20,7 @@ def set_config(
     enable_metadata_routing: None | bool = None,
     skip_parameter_validation: None | bool = None,
 ) -> None: ...
+@contextmanager
 def config_context(
     *,
     assume_finite: None | bool = None,
@@ -31,4 +33,4 @@ def config_context(
     transform_output: None | str = None,
     enable_metadata_routing: None | bool = None,
     skip_parameter_validation: None | bool = None,
-) -> Iterator[None]: ...
+) -> Generator[None, None, None]: ...