improved pyi

Author: flexatone

src/__init__.pyi

@@ -31,6 +31,12 @@ class ErrorInitTypeBlocks(RuntimeError):
     def with_traceback(self, tb: Exception) -> Exception: ...
     def __setstate__(self) -> None: ...
 
+
+class NonUniqueError(RuntimeError):
+    def __init__(self, *args: tp.Any, **kwargs: tp.Any) -> None: ...
+    def with_traceback(self, tb: Exception) -> Exception: ...
+    def __setstate__(self) -> None: ...
+
 class ArrayGO:
     values: np.ndarray
     def __init__(
@@ -110,7 +116,7 @@ class BlockIndex:
 
 
 class FrozenAutoMap:
-    def __init__(self, labels: tp.Iterable[_TLabel] | np.ndarray, /,) -> None: ...
+    def __init__(self, labels: tp.Iterable[_TLabel] | np.ndarray = (), /,) -> None: ...
     def get(self, __key: _TLabel, /,) -> int: ...
     def keys(self) -> tp.Iterator[_TLabel]: ...
     def items(self) -> tp.Iterator[tuple[_TLabel, int]]: ...
@@ -133,9 +139,9 @@ class FrozenAutoMap:
 
 
 class AutoMap(FrozenAutoMap):
-    def __init__(self, labels: tp.Iterable[_TLabel] | np.ndarray, /,) -> None: ...
+    def __init__(self, labels: tp.Iterable[_TLabel] | np.ndarray = (), /,) -> None: ...
     def __ior__(self) -> tp.Any: ...
-    def add(self, __key: int) -> None: ...
+    def add(self, __key: _TLabel) -> None: ...
     def update(self, __keys: tp.Iterable[_TLabel] | np.ndarray) -> None: ...