From 4bd13ff419dac1742767e262181adfb52247880e Mon Sep 17 00:00:00 2001 From: Kaisen Chen Date: Fri, 14 Oct 2022 12:07:38 -0700 Subject: [PATCH] Fix RefResolver's context manager return types (#8898) --- stubs/jsonschema/jsonschema/validators.pyi | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/stubs/jsonschema/jsonschema/validators.pyi b/stubs/jsonschema/jsonschema/validators.pyi index c95dc0bef..5c7d2515d 100644 --- a/stubs/jsonschema/jsonschema/validators.pyi +++ b/stubs/jsonschema/jsonschema/validators.pyi @@ -1,5 +1,6 @@ -from _typeshed import SupportsKeysAndGetItem +from _typeshed import Incomplete, SupportsKeysAndGetItem from collections.abc import Callable, Generator, Iterable, Iterator, Mapping +from contextlib import contextmanager from typing import Any, ClassVar from typing_extensions import TypeAlias @@ -86,8 +87,10 @@ class RefResolver: def resolution_scope(self): ... @property def base_uri(self): ... - def in_scope(self, scope) -> None: ... - def resolving(self, ref) -> None: ... + @contextmanager + def in_scope(self, scope) -> Generator[None, None, None]: ... + @contextmanager + def resolving(self, ref) -> Generator[Incomplete, None, None]: ... def resolve(self, ref): ... def resolve_from_url(self, url): ... def resolve_fragment(self, document, fragment): ...