dycw-utilities 0.125.1__py3-none-any.whl → 0.125.3__py3-none-any.whl

{dycw_utilities-0.125.1.dist-info → dycw_utilities-0.125.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.125.1
+Version: 0.125.3
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

{dycw_utilities-0.125.1.dist-info → dycw_utilities-0.125.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=J1Gf3FyEmeotls_D_lGJIjDWsDDZcqnYOQZ1egNnqYs,60
+utilities/__init__.py,sha256=S9hi5nUNGrmUoeHSNNw9CdOdoY68OUGHaOm0V7chzZo,60
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
 utilities/asyncio.py,sha256=gr2eUx0E6LiCup6VKgUGwh8lAUriGdX2TlK-PZdlvfo,28284
 utilities/atomicwrites.py,sha256=geFjn9Pwn-tTrtoGjDDxWli9NqbYfy3gGL6ZBctiqSo,5393
@@ -24,7 +24,7 @@ utilities/git.py,sha256=wpt5dZ5Oi5931pN24_VLZYaQOvmR0OcQuVtgHzFUN1k,2359
 utilities/hashlib.py,sha256=SVTgtguur0P4elppvzOBbLEjVM3Pea0eWB61yg2ilxo,309
 utilities/http.py,sha256=WcahTcKYRtZ04WXQoWt5EGCgFPcyHD3EJdlMfxvDt-0,946
 utilities/hypothesis.py,sha256=a75izXg9aCBhhDkj_ZgK3TDzlzk38evP8TO7JbYYQvg,46264
-utilities/importlib.py,sha256=ueY3R39hWrUtrVXs39utM2xDig-eyJfYn1FBVxWb3_w,368
+utilities/importlib.py,sha256=mV1xT_O_zt_GnZZ36tl3xOmMaN_3jErDWY54fX39F6Y,429
 utilities/ipython.py,sha256=V2oMYHvEKvlNBzxDXdLvKi48oUq2SclRg5xasjaXStw,763
 utilities/iterables.py,sha256=prKXBdF5QfLTGC-q4567DwO8xzUng_Z-2a4wBkMqyDo,45360
 utilities/jupyter.py,sha256=ft5JA7fBxXKzP-L9W8f2-wbF0QeYc_2uLQNFDVk4Z-M,2917
@@ -35,7 +35,7 @@ utilities/luigi.py,sha256=fpH9MbxJDuo6-k9iCXRayFRtiVbUtibCJKugf7ygpv0,5988
 utilities/math.py,sha256=-mQgbah-dPJwOEWf3SonrFoVZ2AVxMgpeQ3dfVa-oJA,26764
 utilities/memory_profiler.py,sha256=tf2C51P2lCujPGvRt2Rfc7VEw5LDXmVPCG3z_AvBmbU,962
 utilities/modules.py,sha256=iuvLluJya-hvl1Q25-Jk3dLgx2Es3ck4SjJiEkAlVTs,3195
-utilities/more_itertools.py,sha256=fYsGyIPB2s1KRWoH3AkV3CxJxnMLvmidqBf8l1VQnyE,7193
+utilities/more_itertools.py,sha256=tBbjjKx8_Uv_TCjxhPwrGfAx_jRHtvLIZqXVWAsjzqA,8842
 utilities/numpy.py,sha256=Xn23sA2ZbVNqwUYEgNJD3XBYH6IbCri_WkHSNhg3NkY,26122
 utilities/operator.py,sha256=0M2yZJ0PODH47ogFEnkGMBe_cfxwZR02T_92LZVZvHo,3715
 utilities/optuna.py,sha256=loyJGWTzljgdJaoLhP09PT8Jz6o_pwBOwehY33lHkhw,1923
@@ -87,7 +87,7 @@ utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
 utilities/whenever.py,sha256=jS31ZAY5OMxFxLja_Yo5Fidi87Pd-GoVZ7Vi_teqVDA,16743
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=-5j7IQ9nb7gR43rdgA7ms05im-XuqhAk9EJnQBXxCoQ,1874
-dycw_utilities-0.125.1.dist-info/METADATA,sha256=kZ35w8nseblmp9RraacEJcl9ZAIQls6J8FPSiaX3eT0,12852
-dycw_utilities-0.125.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.125.1.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.125.1.dist-info/RECORD,,
+dycw_utilities-0.125.3.dist-info/METADATA,sha256=vMdoTZVKyi-BNSOGgeJYjY6uY-icfbVmvUlAo_SwJmM,12852
+dycw_utilities-0.125.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.125.3.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.125.3.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.125.1"
+__version__ = "0.125.3"

utilities/importlib.py

@@ -4,11 +4,13 @@ from importlib import import_module
 from importlib.util import find_spec
 
 
-def is_valid_import(module: str, name: str) -> bool:
+def is_valid_import(module: str, /, *, name: str | None = None) -> bool:
     """Check if an import is valid."""
     spec = find_spec(module)
     if spec is None:
         return False
+    if name is None:
+        return True
     mod = import_module(module)
     return hasattr(mod, name)
 

utilities/more_itertools.py

@@ -21,12 +21,14 @@ from more_itertools import bucket, partition, split_into
 from more_itertools import peekable as _peekable
 
 from utilities.functions import get_class_name
+from utilities.iterables import OneNonUniqueError, one
+from utilities.reprlib import get_repr
 from utilities.sentinel import Sentinel, sentinel
+from utilities.types import THashable
 
 if TYPE_CHECKING:
     from collections.abc import Callable, Iterable, Iterator, Mapping, Sequence
 
-    from utilities.types import THashable
 
 _T = TypeVar("_T")
 _U = TypeVar("_U")
@@ -35,6 +37,26 @@ _U = TypeVar("_U")
 ##
 
 
+@overload
+def bucket_mapping(
+    iterable: Iterable[_T],
+    func: Callable[[_T], THashable],
+    /,
+    *,
+    transform: Callable[[_T], _U],
+    list: bool = False,
+    unique: Literal[True],
+) -> Mapping[THashable, _U]: ...
+@overload
+def bucket_mapping(
+    iterable: Iterable[_T],
+    func: Callable[[_T], THashable],
+    /,
+    *,
+    transform: Callable[[_T], _U] | None = None,
+    list: bool = False,
+    unique: Literal[True],
+) -> Mapping[THashable, _T]: ...
 @overload
 def bucket_mapping(
     iterable: Iterable[_T],
@@ -79,11 +101,14 @@ def bucket_mapping(
     *,
     transform: Callable[[_T], _U] | None = None,
     list: bool = False,
+    unique: bool = False,
 ) -> (
     Mapping[THashable, Iterator[_T]]
     | Mapping[THashable, Iterator[_U]]
     | Mapping[THashable, Sequence[_T]]
     | Mapping[THashable, Sequence[_U]]
+    | Mapping[THashable, _T]
+    | Mapping[THashable, _U]
 ): ...
 def bucket_mapping(
     iterable: Iterable[_T],
@@ -92,26 +117,55 @@ def bucket_mapping(
     *,
     transform: Callable[[_T], _U] | None = None,
     list: bool = False,  # noqa: A002
+    unique: bool = False,
 ) -> (
     Mapping[THashable, Iterator[_T]]
     | Mapping[THashable, Iterator[_U]]
     | Mapping[THashable, Sequence[_T]]
     | Mapping[THashable, Sequence[_U]]
+    | Mapping[THashable, _T]
+    | Mapping[THashable, _U]
 ):
     """Bucket the values of iterable into a mapping."""
     b = bucket(iterable, func)
     mapping = {key: b[key] for key in b}
     match transform, list:
         case None, False:
-            return mapping
+            ...
         case None, True:
-            return {k: builtins.list(v) for k, v in mapping.items()}
+            mapping = {k: builtins.list(v) for k, v in mapping.items()}
         case _, False:
-            return {k: map(transform, v) for k, v in mapping.items()}
+            mapping = {k: map(transform, v) for k, v in mapping.items()}
         case _, True:
-            return {k: builtins.list(map(transform, v)) for k, v in mapping.items()}
+            mapping = {k: builtins.list(map(transform, v)) for k, v in mapping.items()}
         case _ as never:
             assert_never(never)
+    if not unique:
+        return mapping
+    results = {}
+    error_no_transform: dict[THashable, tuple[_T, _T]] = {}
+    for key, value in mapping.items():
+        try:
+            results[key] = one(value)
+        except OneNonUniqueError as error:
+            error_no_transform[key] = (error.first, error.second)
+    if len(error_no_transform) >= 1:
+        raise BucketMappingError(errors=error_no_transform)
+    return results
+
+
+@dataclass(kw_only=True, slots=True)
+class BucketMappingError(Exception, Generic[THashable, _U]):
+    errors: Mapping[THashable, tuple[_U, _U]]
+
+    @override
+    def __str__(self) -> str:
+        parts = [
+            f"{get_repr(key)} (#1: {get_repr(first)}, #2: {get_repr(second)})"
+            for key, (first, second) in self.errors.items()
+        ]
+        desc = ", ".join(parts)
+        return f"Buckets must contain exactly one item each; got {desc}"
 
 
 ##
@@ -264,6 +318,7 @@ def _yield_splits3(
 
 
 __all__ = [
+    "BucketMappingError",
     "Split",
     "bucket_mapping",
     "partition_list",