X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=dict_utils.py;h=79c86edf286f2c9ea9983906385365199be74892;hb=36fea7f15ed17150691b5b3ead75450e575229ef;hp=7b0edb50a87a5854f20326408cfc0a03d22408dc;hpb=d08bad64a6884f25d28a2c38c6cd1c87b4335188;p=python_utils.git

diff --git a/dict_utils.py b/dict_utils.py
index 7b0edb5..79c86ed 100644
--- a/dict_utils.py
+++ b/dict_utils.py
@@ -9,7 +9,7 @@ def init_or_inc(
     key: Any,
     *,
     init_value: Any = 1,
-    inc_function: Callable[..., Any] = lambda x: x + 1
+    inc_function: Callable[..., Any] = lambda x: x + 1,
 ) -> bool:
     """
     Initialize a dict value (if it doesn't exist) or increments it (using the
@@ -38,6 +38,7 @@ def shard(d: Dict[Any, Any], size: int) -> Iterator[Dict[Any, Any]]:
     """
     Shards a dict into N subdicts which, together, contain all keys/values
     from the original unsharded dict.
+
     """
     items = d.items()
     for x in range(0, len(d), size):
@@ -46,6 +47,7 @@ def shard(d: Dict[Any, Any], size: int) -> Iterator[Dict[Any, Any]]:
 
 def coalesce_by_creating_list(key, new_value, old_value):
     from list_utils import flatten
+
     return flatten([new_value, old_value])
 
 
@@ -66,9 +68,9 @@ def raise_on_duplicated_keys(key, new_value, old_value):
 
 
 def coalesce(
-        inputs: Iterator[Dict[Any, Any]],
-        *,
-        aggregation_function: Callable[[Any, Any], Any] = coalesce_by_creating_list
+    inputs: Iterator[Dict[Any, Any]],
+    *,
+    aggregation_function: Callable[[Any, Any], Any] = coalesce_by_creating_list,
 ) -> Dict[Any, Any]:
     """Merge N dicts into one dict containing the union of all keys /
     values in the input dicts.  When keys collide, apply the
@@ -196,7 +198,9 @@ def min_key(d: Dict[Any, Any]) -> Any:
     return min(d.keys())
 
 
-def parallel_lists_to_dict(keys: List[Any], values: List[Any]) -> Dict[Any, Any]:
+def parallel_lists_to_dict(
+    keys: List[Any], values: List[Any]
+) -> Dict[Any, Any]:
     """Given two parallel lists (keys and values), create and return
     a dict.
 
@@ -207,7 +211,9 @@ def parallel_lists_to_dict(keys: List[Any], values: List[Any]) -> Dict[Any, Any]
 
     """
     if len(keys) != len(values):
-        raise Exception("Parallel keys and values lists must have the same length")
+        raise Exception(
+            "Parallel keys and values lists must have the same length"
+        )
     return dict(zip(keys, values))
 
 
@@ -230,4 +236,5 @@ def dict_to_key_value_lists(d: Dict[Any, Any]) -> Tuple[List[Any], List[Any]]:
 
 if __name__ == '__main__':
     import doctest
+
     doctest.testmod()