relib 1.0.5__py3-none-any.whl → 1.0.7__py3-none-any.whl

relib/__init__.py

@@ -2,7 +2,8 @@ from .utils import (
   list_split,
   drop_none,
   distinct,
-  find,
+  first,
+  move_value,
   transpose_dict,
   make_combinations_by_dict,
   merge_dicts,

relib/utils.py

@@ -1,4 +1,4 @@
-from typing import TypeVar, Union, Iterable, Callable
+from typing import TypeVar, Union, Iterable, Callable, Any, cast, overload
 from itertools import chain
 import numpy as np
 import re
@@ -6,6 +6,11 @@ import re
 T = TypeVar('T')
 U = TypeVar('U')
 K = TypeVar('K')
+K1, K2, K3, K4, K5, K6 = TypeVar('K1'), TypeVar('K2'), TypeVar('K3'), TypeVar('K4'), TypeVar('K5'), TypeVar('K6')
+
+def non_none(obj: Union[T, None]) -> T:
+  assert obj is not None
+  return obj
 
 def list_split(l: list[T], sep: T) -> list[list[T]]:
   l = [sep, *l, sep]
@@ -22,9 +27,14 @@ def drop_none(l: Iterable[Union[T, None]]) -> list[T]:
 def distinct(items: Iterable[T]) -> list[T]:
   return list(set(items))
 
-def find(iterable: Iterable[T]) -> Union[T, None]:
+def first(iterable: Iterable[T]) -> Union[T, None]:
   return next(iter(iterable), None)
 
+def move_value(l: Iterable[T], from_i: int, to_i: int) -> list[T]:
+  l = list(l)
+  l.insert(to_i, l.pop(from_i))
+  return l
+
 def transpose_dict(des):
   if isinstance(des, list):
     keys = list(des[0].keys()) if des else []
@@ -55,10 +65,10 @@ def make_combinations_by_dict(des, keys=None, pairs=[]):
   ])
 
 def merge_dicts(*dicts: dict[K, T]) -> dict[K, T]:
-  output = {}
+  result = {}
   for d in dicts:
-    output.update(d)
-  return output
+    result.update(d)
+  return result
 
 def intersect(*lists: Iterable[T]) -> list[T]:
   return list(set.intersection(*map(set, lists)))
@@ -97,7 +107,25 @@ def transpose(tuples, default_num_returns=0):
 def map_dict(fn: Callable[[T], U], d: dict[K, T]) -> dict[K, U]:
   return {key: fn(value) for key, value in d.items()}
 
-def deepen_dict(d):
+@overload
+def deepen_dict(d: dict[tuple[K1], U]) -> dict[K1, U]: ...
+
+@overload
+def deepen_dict(d: dict[tuple[K1, K2], U]) -> dict[K1, dict[K2, U]]: ...
+
+@overload
+def deepen_dict(d: dict[tuple[K1, K2, K3], U]) -> dict[K1, dict[K2, dict[K3, U]]]: ...
+
+@overload
+def deepen_dict(d: dict[tuple[K1, K2, K3, K4], U]) -> dict[K1, dict[K2, dict[K3, dict[K4, U]]]]: ...
+
+@overload
+def deepen_dict(d: dict[tuple[K1, K2, K3, K4, K5], U]) -> dict[K1, dict[K2, dict[K3, dict[K4, dict[K5, U]]]]]: ...
+
+@overload
+def deepen_dict(d: dict[tuple[K1, K2, K3, K4, K5, K6], U]) -> dict[K1, dict[K2, dict[K3, dict[K4, dict[K5, dict[K6, U]]]]]]: ...
+
+def deepen_dict(d: dict[tuple[Any, ...], Any]) -> dict:
   output = {}
   for (*tail, head), value in d.items():
     curr = output
@@ -116,13 +144,13 @@ def group(pairs: Iterable[tuple[K, T]]) -> dict[K, list[T]]:
     values_by_key[key].append(value)
   return values_by_key
 
-def get_at(d, keys, default):
+def get_at(d: dict, keys: Iterable[Any], default: T) -> T:
   try:
     for key in keys:
       d = d[key]
   except KeyError:
     return default
-  return d
+  return cast(Any, d)
 
 def sized_partitions(values: Iterable[T], part_size: int) -> list[list[T]]:
   if not isinstance(values, list):
@@ -137,7 +165,7 @@ def num_partitions(values: Iterable[T], num_parts: int) -> list[list[T]]:
   return [values[i * part_size:(i + 1) * part_size] for i in range(num_parts)]
 
 def _cat_tile(cats, n_tile):
-    return cats[np.tile(np.arange(len(cats)), n_tile)]
+  return cats[np.tile(np.arange(len(cats)), n_tile)]
 
 def df_from_array(
   value_cols: dict[str, np.ndarray],
@@ -148,15 +176,14 @@ def df_from_array(
   dim_names = [name for name, _ in dim_labels]
   dim_sizes = np.array([len(labels) for _, labels in dim_labels])
   assert all(array.shape == tuple(dim_sizes) for array in value_cols.values())
-  repeats_tiles = [
+  array_offsets = [
     (dim_sizes[i + 1:].prod(), dim_sizes[:i].prod())
     for i in range(len(dim_sizes))
   ]
-  label_arrays = [
-    _cat_tile(pd.Categorical(labels).repeat(repeats), tiles)
-    for labels, (repeats, tiles) in zip(dim_labels, repeats_tiles)
-  ]
-  category_cols = dict(zip(dim_names, label_arrays))
+  category_cols = {
+    dim: _cat_tile(pd.Categorical(labels).repeat(repeats), tiles)
+    for dim, labels, (repeats, tiles) in zip(dim_names, dim_labels, array_offsets)
+  }
   value_cols = {name: array.reshape(-1) for name, array in value_cols.items()}
   df = pd.DataFrame({**category_cols, **value_cols}, copy=False)
   if indexed:

{relib-1.0.5.dist-info → relib-1.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: relib
-Version: 1.0.5
+Version: 1.0.7
 Home-page: https://github.com/Reddan/relib
 Author: Hampus Hallman
 Author-email: me@hampushallman.com

relib-1.0.7.dist-info/RECORD

@@ -0,0 +1,10 @@
+relib/__init__.py,sha256=4yr8xPi3VMbiFlApussB4OXU_U6wzhje06qD0Ad3Gq4,471
+relib/hashing.py,sha256=6iAPRiJI_4jaSooZRFJnqK2limXqTmErzcwpd050LAA,8943
+relib/measure_duration.py,sha256=jJa5Kh5FxaBysS__nkwqcnTt8Uc2niLucXfTzFE0j-E,555
+relib/raypipe.py,sha256=ynEoXs1dnD-360_uQC8v89xjiilt3knpocXpFaQ3plA,1905
+relib/utils.py,sha256=II0PikMmpc9Ds2F3jPRXuIFD5RTUk7Bk2-8C4gvL0T8,6271
+relib-1.0.7.dist-info/LICENSE.txt,sha256=t9LfkVbmcvZjP0x3Sq-jR38UfTNbNtRQvc0Q8HWmLak,1054
+relib-1.0.7.dist-info/METADATA,sha256=Lax2ex1ap8pZJrXzKA4Dc5qnSBt2FqHOS5XK51WJNQQ,260
+relib-1.0.7.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+relib-1.0.7.dist-info/top_level.txt,sha256=Yc96FwkbRYj4AQVatga8uK4hH9ATKI9XIyEH_1ba6KQ,6
+relib-1.0.7.dist-info/RECORD,,

{relib-1.0.5.dist-info → relib-1.0.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.40.0)
+Generator: setuptools (70.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

relib-1.0.5.dist-info/RECORD

@@ -1,10 +0,0 @@
-relib/__init__.py,sha256=Zm6aMrr9A70laX2otqq60JLSO9bLiiwIQQJmk2Qzp_k,456
-relib/hashing.py,sha256=6iAPRiJI_4jaSooZRFJnqK2limXqTmErzcwpd050LAA,8943
-relib/measure_duration.py,sha256=jJa5Kh5FxaBysS__nkwqcnTt8Uc2niLucXfTzFE0j-E,555
-relib/raypipe.py,sha256=ynEoXs1dnD-360_uQC8v89xjiilt3knpocXpFaQ3plA,1905
-relib/utils.py,sha256=Gh_Mb1z8FyzsPAsdCiNLUskztr16ajSrvzQ_lvAn_Tk,5258
-relib-1.0.5.dist-info/LICENSE.txt,sha256=t9LfkVbmcvZjP0x3Sq-jR38UfTNbNtRQvc0Q8HWmLak,1054
-relib-1.0.5.dist-info/METADATA,sha256=a8d2h8VbXerVegSy2fq3gpWhyHubrfs5UVQKz7Ybqls,260
-relib-1.0.5.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-relib-1.0.5.dist-info/top_level.txt,sha256=Yc96FwkbRYj4AQVatga8uK4hH9ATKI9XIyEH_1ba6KQ,6
-relib-1.0.5.dist-info/RECORD,,