allow object type in dict.get and dict.pop

152fee56ae355c7bc42fd9dea33b43fbd2ea76aa
commit 152fee56ae355c7bc42fd9dea33b43fbd2ea76aa
diff --git a/stdlib/@tests/test_cases/builtins/check_dict.py b/stdlib/@tests/test_cases/builtins/check_dict.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import os
-from typing import Any, Dict, Generic, Iterable, Mapping, TypeVar, Union
+from typing import Any, Dict, Generic, Iterable, Literal, Mapping, TypeVar, Union
 from typing_extensions import Self, assert_type
 
 ###################################################################
@@ -103,6 +103,16 @@ def test_iterable_tuple_overload(x: Iterable[tuple[int, str]]) -> dict[int, str]
 result = d_str.get("key", int_value)  # type: ignore[arg-type]
 
 
+def test_dict_signature_overlapping_type(d: dict[Literal["foo", "bar"], int], key: str) -> None:
+    # Note: annotations also allow using keys of a disjoint type (e.g., int),
+    #   linters / type checkers are free to issue warnings in such cases.
+    #   statically, a .get(arg) is superfluous if the intersection of the
+    #   dict key type and the argument type is empty.
+    #   So we only test a case with non-empty intersection here.
+    d.get(key)
+    d.pop(key)
+
+
 # Return values also make things weird
 
 # Pyright doesn't have a version of no-any-return,

diff --git a/stdlib/builtins.pyi b/stdlib/builtins.pyi
@@ -1336,19 +1336,23 @@ class dict(MutableMapping[_KT, _VT]):
     def fromkeys(cls, iterable: Iterable[_T], value: _S, /) -> dict[_T, _S]: ...
 
     # Positional-only in dict, but not in MutableMapping
+    # dict.get allows arbitrary types, which matches runtime semantics.
+    # linters may choose to warn if the given type does not overlap with the key type
     @overload  # type: ignore[override]
-    def get(self, key: _KT, default: None = None, /) -> _VT | None: ...
+    def get(self, key: object, default: None = None, /) -> _VT | None: ...
     @overload
-    def get(self, key: _KT, default: _VT, /) -> _VT: ...
+    def get(self, key: object, default: _VT, /) -> _VT: ...
     @overload
-    def get(self, key: _KT, default: _T, /) -> _VT | _T: ...
+    def get(self, key: object, default: _T, /) -> _VT | _T: ...
 
+    # dict.pop allows arbitrary types, which matches runtime semantics.
+    # linters may choose to warn if the given type does not overlap with the key type
     @overload
-    def pop(self, key: _KT, /) -> _VT: ...
+    def pop(self, key: object, /) -> _VT: ...
     @overload
-    def pop(self, key: _KT, default: _VT, /) -> _VT: ...
+    def pop(self, key: object, default: _VT, /) -> _VT: ...
     @overload
-    def pop(self, key: _KT, default: _T, /) -> _VT | _T: ...
+    def pop(self, key: object, default: _T, /) -> _VT | _T: ...
 
     def __len__(self) -> int: ...
     def __getitem__(self, key: _KT, /) -> _VT: ...

diff --git a/stdlib/collections/__init__.pyi b/stdlib/collections/__init__.pyi
@@ -406,11 +406,11 @@ class OrderedDict(dict[_KT, _VT]):
 
     # Same as dict.pop, but accepts keyword arguments
     @overload
-    def pop(self, key: _KT) -> _VT: ...
+    def pop(self, key: object) -> _VT: ...
     @overload
-    def pop(self, key: _KT, default: _VT) -> _VT: ...
+    def pop(self, key: object, default: _VT) -> _VT: ...
     @overload
-    def pop(self, key: _KT, default: _T) -> _VT | _T: ...
+    def pop(self, key: object, default: _T) -> _VT | _T: ...
 
     def __eq__(self, value: object, /) -> bool: ...
 

diff --git a/stdlib/mailbox.pyi b/stdlib/mailbox.pyi
@@ -113,7 +113,7 @@ class Mailbox(Generic[_MessageT_co]):
     def clear(self) -> None: ...
 
     @overload
-    def pop(self, key: str, default: None = None) -> _MessageT_co | None: ...
+    def pop(self, key: str) -> _MessageT_co | None: ...
     @overload
     def pop(self, key: str, default: _T) -> _MessageT_co | _T: ...
 

diff --git a/stubs/boltons/boltons/cacheutils.pyi b/stubs/boltons/boltons/cacheutils.pyi
@@ -34,9 +34,9 @@ class LRI(dict[_KT, _VT]):
     def __delitem__(self, key: _KT) -> None: ...
 
     @overload
-    def pop(self, key: _KT) -> _VT: ...
+    def pop(self, key: object) -> _VT: ...
     @overload
-    def pop(self, key: _KT, default: _T) -> _T | _VT: ...
+    def pop(self, key: object, default: _T) -> _T | _VT: ...
 
     def popitem(self) -> tuple[_KT, _VT]: ...
     def clear(self) -> None: ...

diff --git a/stubs/boltons/boltons/dictutils.pyi b/stubs/boltons/boltons/dictutils.pyi
@@ -69,7 +69,7 @@ class OneToOne(dict[_KT, _VT]):
     inv: OneToOne[_VT, _KT]
     def clear(self) -> None: ...
     def copy(self) -> Self: ...
-    def pop(self, key: _KT, default: _VT | _T = ...) -> _VT | _T: ...
+    def pop(self, key: object, default: _VT | _T = ...) -> _VT | _T: ...
     def popitem(self) -> tuple[_KT, _VT]: ...
 
     @overload

diff --git a/stubs/yt-dlp/yt_dlp/utils/networking.pyi b/stubs/yt-dlp/yt_dlp/utils/networking.pyi
@@ -23,11 +23,11 @@ class HTTPHeaderDict(dict[str, str]):
     def get(self, key: str, /, default: type[NO_DEFAULT] | _T = ...) -> str | _T | type[NO_DEFAULT]: ...
 
     @overload
-    def pop(self, key: str, /) -> str: ...
+    def pop(self, key: object, /) -> str: ...
     @overload
-    def pop(self, key: str, /, default: _T) -> str | _T: ...
+    def pop(self, key: object, /, default: _T) -> str | _T: ...
     @overload
-    def pop(self, key: str, /, default: type[NO_DEFAULT] | _T | str = ...) -> str | _T | type[NO_DEFAULT]: ...
+    def pop(self, key: object, /, default: type[NO_DEFAULT] | _T | str = ...) -> str | _T | type[NO_DEFAULT]: ...
 
     @overload
     def setdefault(self, key: str, /) -> str: ...