relib 1.0.3__py3-none-any.whl → 1.0.4__py3-none-any.whl

relib/__init__.py

@@ -9,6 +9,7 @@ from .utils import (
   intersect,
   ensure_tuple,
   omit,
+  pick,
   dict_by,
   tuple_by,
   flatten,

relib/utils.py

@@ -73,6 +73,9 @@ def omit(d: dict[K, T], keys: Iterable[K]) -> dict[K, T]:
       del d[key]
   return d
 
+def pick(d: dict[K, T], keys: Iterable[K]) -> dict[K, T]:
+  return {key: d[key] for key in keys}
+
 def dict_by(keys: Iterable[K], values: Iterable[T]) -> dict[K, T]:
   return dict(zip(keys, values))
 
@@ -132,6 +135,9 @@ def num_partitions(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)]
 
+def _cat_tile(cats, n_tile):
+    return cats[np.tile(np.arange(len(cats)), n_tile)]
+
 def df_from_array(
   value_cols: dict[str, np.ndarray],
   dim_labels: list[tuple[str, list[Union[str, int, float]]]],
@@ -145,7 +151,7 @@ def df_from_array(
     for i in range(len(dim_sizes))
   ]
   label_arrays = [
-    np.tile(np.repeat(pd.Categorical(labels), repeats), tiles)
+    _cat_tile(pd.Categorical(labels).repeat(repeats), tiles)
     for labels, (repeats, tiles) in zip(dim_labels, repeats_tiles)
   ]
   df = pd.DataFrame(dict(zip(dim_names, label_arrays)))

{relib-1.0.3.dist-info → relib-1.0.4.dist-info}/METADATA

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

relib-1.0.4.dist-info/RECORD

@@ -0,0 +1,10 @@
+relib/__init__.py,sha256=d7vbtHV079on1ZRQeT5SAxNHtfSd14JDN1fHCGbbsjE,446
+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=V5pdaO2qZWjOkr__eMbFfF2X4mdk9b2THQ8Ve26UJcU,5221
+relib-1.0.4.dist-info/LICENSE.txt,sha256=t9LfkVbmcvZjP0x3Sq-jR38UfTNbNtRQvc0Q8HWmLak,1054
+relib-1.0.4.dist-info/METADATA,sha256=lzPhG9j-DBJ_X6-camK9N7pAoTp8dLJ6V-e3GQWsPpI,260
+relib-1.0.4.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+relib-1.0.4.dist-info/top_level.txt,sha256=Yc96FwkbRYj4AQVatga8uK4hH9ATKI9XIyEH_1ba6KQ,6
+relib-1.0.4.dist-info/RECORD,,

relib-1.0.3.dist-info/RECORD

@@ -1,10 +0,0 @@
-relib/__init__.py,sha256=pUJoJ4ctBeistvdRn_jrQUDAVXsBVwBf_0LdcbCkkuc,438
-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=pd4iigtwQah2kAQnwhBJ4VfGlsiXfY-geP1Ormws-aQ,5040
-relib-1.0.3.dist-info/LICENSE.txt,sha256=t9LfkVbmcvZjP0x3Sq-jR38UfTNbNtRQvc0Q8HWmLak,1054
-relib-1.0.3.dist-info/METADATA,sha256=8SDVU5Tu5LoEtThzqtrE78PX_D9UCZ8gxr6AXi2RWNA,260
-relib-1.0.3.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-relib-1.0.3.dist-info/top_level.txt,sha256=Yc96FwkbRYj4AQVatga8uK4hH9ATKI9XIyEH_1ba6KQ,6
-relib-1.0.3.dist-info/RECORD,,