use Hashable type as dict key (instead of Any)

adr26 · Jun 10, 2021 · 840ba92 · 840ba92
1 parent a949af1
commit 840ba92
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/bitarray/__init__.pyi b/bitarray/__init__.pyi
@@ -108,7 +108,7 @@ class bitarray:
 
 
 class frozenbitarray(bitarray):
-    ...
+    def __hash__(self) -> int: ...
 
 
 def bits2bytes(n: int, /) -> int: ...

diff --git a/bitarray/util.pyi b/bitarray/util.pyi
@@ -1,4 +1,4 @@
-from typing import Any, AnyStr, BinaryIO, Literal, Optional, Union
+from typing import Any, AnyStr, BinaryIO, Hashable, Literal, Optional, Union
 
 from bitarray import bitarray
 
@@ -41,5 +41,5 @@ def int2ba(i: int, /,
 
 def serialize(a: bitarray, /) -> bytes: ...
 def deserialize(b: bytes, /) -> bitarray: ...
-def huffman_code(freq_map: dict[Any, Union[int, float]],
-                 endian: Optional[str] = ...) -> dict[Any, bitarray]: ...
+def huffman_code(freq_map: dict[Hashable, Union[int, float]],
+                 endian: Optional[str] = ...) -> dict[Hashable, bitarray]: ...
-Original file line number
+Diff line change
@@ Expand Up / @@ -108,7 +108,7 @@ class bitarray: @@
     class frozenbitarray(bitarray):
-        ...
+        def __hash__(self) -> int: ...
     def bits2bytes(n: int, /) -> int: ...
@@ Expand Down @@