relib 1.3.0__tar.gz → 1.3.1__tar.gz

{relib-1.3.0 → relib-1.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: relib
-Version: 1.3.0
+Version: 1.3.1
 Project-URL: Repository, https://github.com/Reddan/relib.git
 Author: Hampus Hallman
 License: Copyright 2018-2025 Hampus Hallman

{relib-1.3.0 → relib-1.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "relib"
-version = "1.3.0"
+version = "1.3.1"
 requires-python = ">=3.12"
 dependencies = []
 authors = [

{relib-1.3.0 → relib-1.3.1}/relib/dict_utils.py

@@ -52,9 +52,8 @@ def get_at[T](d: dict, keys: Iterable[Any], default: T) -> T:
 
 def dict_firsts[T, K](pairs: Iterable[tuple[K, T]]) -> dict[K, T]:
   result: dict[K, T] = {}
-  for key, item in pairs:
-    if key not in result:
-      result[key] = item
+  for key, value in pairs:
+    result.setdefault(key, value)
   return result
 
 def group[T, K](pairs: Iterable[tuple[K, T]]) -> dict[K, list[T]]:
@@ -63,6 +62,16 @@ def group[T, K](pairs: Iterable[tuple[K, T]]) -> dict[K, list[T]]:
     values_by_key.setdefault(key, []).append(value)
   return values_by_key
 
+def flatten_dict_inner(d, prefix=()):
+  for key, value in d.items():
+    if not isinstance(value, dict) or value == {}:
+      yield prefix + (key,), value
+    else:
+      yield from flatten_dict_inner(value, prefix + (key,))
+
+def flatten_dict(deep_dict: dict, prefix=()) -> dict:
+  return dict(flatten_dict_inner(deep_dict, prefix))
+
 @overload
 def deepen_dict[K1, U](d: dict[tuple[K1], U]) -> dict[K1, U]: ...
 @overload
@@ -85,13 +94,3 @@ def deepen_dict(d: dict[tuple[Any, ...], Any]) -> dict:
       curr = curr.setdefault(key, {})
     curr[head] = value
   return output
-
-def flatten_dict_inner(d, prefix=()):
-  for key, value in d.items():
-    if not isinstance(value, dict) or value == {}:
-      yield prefix + (key,), value
-    else:
-      yield from flatten_dict_inner(value, prefix + (key,))
-
-def flatten_dict(deep_dict: dict, prefix=()) -> dict:
-  return dict(flatten_dict_inner(deep_dict, prefix))

{relib-1.3.0 → relib-1.3.1}/relib/iter_utils.py

@@ -5,7 +5,7 @@ from .dict_utils import dict_firsts
 __all__ = [
   "distinct_by", "distinct", "drop_none",
   "first", "flatten",
-  "intersect",
+  "interleave", "intersect",
   "list_split",
   "move_value",
   "num_partitions",
@@ -36,27 +36,13 @@ def move_value[T](iterable: Iterable[T], from_i: int, to_i: int) -> list[T]:
   return values
 
 def reversed_enumerate[T](values: list[T] | tuple[T, ...]) -> Iterable[tuple[int, T]]:
-  return zip(reversed(range(len(values))), reversed(values))
+  return zip(range(len(values))[::1], reversed(values))
 
 def intersect[T](*iterables: Iterable[T]) -> list[T]:
   return list(set.intersection(*map(set, iterables)))
 
-@overload
-def flatten[T](iterable: Iterable[T], depth: Literal[0]) -> list[T]: ...
-@overload
-def flatten[T](iterable: Iterable[Iterable[T]], depth: Literal[1] = 1) -> list[T]: ...
-@overload
-def flatten[T](iterable: Iterable[Iterable[Iterable[T]]], depth: Literal[2]) -> list[T]: ...
-@overload
-def flatten[T](iterable: Iterable[Iterable[Iterable[Iterable[T]]]], depth: Literal[3]) -> list[T]: ...
-@overload
-def flatten[T](iterable: Iterable[Iterable[Iterable[Iterable[Iterable[T]]]]], depth: Literal[4]) -> list[T]: ...
-@overload
-def flatten(iterable: Iterable, depth: int) -> list: ...
-def flatten(iterable: Iterable, depth: int = 1) -> list:
-  for _ in range(depth):
-    iterable = chain.from_iterable(iterable)
-  return list(iterable)
+def interleave[T](*iterables: Iterable[T]) -> list[T]:
+  return flatten(zip(*iterables))
 
 def list_split[T](iterable: Iterable[T], sep: T) -> list[list[T]]:
   values = [sep, *iterable, sep]
@@ -77,6 +63,23 @@ def num_partitions[T](values: Iterable[T], num_parts: int) -> list[list[T]]:
   part_size = (len(values) / num_parts).__ceil__()
   return [values[i * part_size:(i + 1) * part_size] for i in range(num_parts)]
 
+@overload
+def flatten[T](iterable: Iterable[T], depth: Literal[0]) -> list[T]: ...
+@overload
+def flatten[T](iterable: Iterable[Iterable[T]], depth: Literal[1] = 1) -> list[T]: ...
+@overload
+def flatten[T](iterable: Iterable[Iterable[Iterable[T]]], depth: Literal[2]) -> list[T]: ...
+@overload
+def flatten[T](iterable: Iterable[Iterable[Iterable[Iterable[T]]]], depth: Literal[3]) -> list[T]: ...
+@overload
+def flatten[T](iterable: Iterable[Iterable[Iterable[Iterable[Iterable[T]]]]], depth: Literal[4]) -> list[T]: ...
+@overload
+def flatten(iterable: Iterable, depth: int) -> list: ...
+def flatten(iterable: Iterable, depth: int = 1) -> list:
+  for _ in range(depth):
+    iterable = chain.from_iterable(iterable)
+  return list(iterable)
+
 @overload
 def transpose[T1, T2](tuples: Iterable[tuple[T1, T2]], default_num_returns=0) -> tuple[list[T1], list[T2]]: ...
 @overload

{relib-1.3.0 → relib-1.3.1}/relib/runtime_tools.py

@@ -10,6 +10,7 @@ from .processing_utils import noop
 __all__ = [
   "as_async", "async_limit",
   "clear_console", "console_link",
+  "default_executor", "default_workers",
   "roll_tasks",
   "measure_duration",
 ]
@@ -18,6 +19,7 @@ P = ParamSpec("P")
 R = TypeVar("R")
 
 default_workers = min(32, (os.cpu_count() or 1) + 4)
+default_executor = ThreadPoolExecutor(max_workers=default_workers)
 
 def clear_console() -> None:
   os.system("cls" if os.name == "nt" else "clear")
@@ -42,8 +44,8 @@ async def roll_tasks[T](tasks: Iterable[Awaitable[T]], workers=default_workers,
     update = partial(pbar.update, 1)
     return await asyncio.gather(*[worker(task, semaphore, update) for task in tasks])
 
-def as_async(workers=default_workers) -> Callable[[Callable[P, R]], Callable[P, Awaitable[R]]]:
-  executor = ThreadPoolExecutor(max_workers=workers)
+def as_async(workers: int | ThreadPoolExecutor = default_executor) -> Callable[[Callable[P, R]], Callable[P, Awaitable[R]]]:
+  executor = ThreadPoolExecutor(max_workers=workers) if isinstance(workers, int) else workers
 
   def on_fn(func: Callable[P, R]) -> Callable[P, Awaitable[R]]:
     @wraps(func)
@@ -79,7 +81,7 @@ class measure_duration:
   def __exit__(self, *_):
     duration = round(time() - self.start, 4)
     depth = len(active_mds) - 1
-    indent = ('──' * depth) + (' ' * (depth > 0))
-    text = '{}: {} seconds'.format(self.name, duration)
+    indent = "──" * depth + " " * (depth > 0)
+    text = f"{self.name}: {duration} seconds"
     print(indent + text)
     active_mds.remove(self)

{relib-1.3.0 → relib-1.3.1}/uv.lock

@@ -106,7 +106,7 @@ wheels = [
 
 [[package]]
 name = "relib"
-version = "1.3.0"
+version = "1.3.1"
 source = { editable = "." }
 
 [package.dev-dependencies]