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

relib/__init__.py

@@ -1,4 +1,5 @@
 from .utils import (
+  non_none,
   list_split,
   drop_none,
   distinct,

relib/measure_duration.py

@@ -6,16 +6,15 @@ active_mds = []
 class measure_duration:
   def __init__(self, name):
     self.name = name
-    self.start = time()
     active_mds.append(self)
 
   def __enter__(self):
-    pass
+    self.start = time()
 
   def __exit__(self, *_):
     duration = round(time() - self.start, 4)
-    indent_level = len(active_mds) - 1
-    indentatation = ('──' * indent_level) + (' ' * (indent_level > 0))
+    depth = len(active_mds) - 1
+    indent = ('──' * depth) + (' ' * (depth > 0))
     text = '{}: {} seconds'.format(self.name, duration)
-    print(colored(indentatation + text, attrs=['dark']))
+    print(colored(indent + text, attrs=['dark']))
     active_mds.remove(self)

relib/utils.py

@@ -1,4 +1,4 @@
-from typing import TypeVar, Union, Iterable, Callable, Any, cast, overload
+from typing import TypeVar, Iterable, Callable, Any, cast, overload
 from itertools import chain
 import numpy as np
 import re
@@ -8,7 +8,7 @@ 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:
+def non_none(obj: T | None) -> T:
   assert obj is not None
   return obj
 
@@ -21,13 +21,13 @@ def list_split(l: list[T], sep: T) -> list[list[T]]:
     for start, end in ranges
   ]
 
-def drop_none(l: Iterable[Union[T, None]]) -> list[T]:
+def drop_none(l: Iterable[T | None]) -> list[T]:
   return [x for x in l if x is not None]
 
 def distinct(items: Iterable[T]) -> list[T]:
   return list(set(items))
 
-def first(iterable: Iterable[T]) -> Union[T, None]:
+def first(iterable: Iterable[T]) -> T | None:
   return next(iter(iterable), None)
 
 def move_value(l: Iterable[T], from_i: int, to_i: int) -> list[T]:
@@ -65,6 +65,8 @@ def make_combinations_by_dict(des, keys=None, pairs=[]):
   ])
 
 def merge_dicts(*dicts: dict[K, T]) -> dict[K, T]:
+  if len(dicts) == 1:
+    return dicts[0]
   result = {}
   for d in dicts:
     result.update(d)
@@ -73,7 +75,7 @@ def merge_dicts(*dicts: dict[K, T]) -> dict[K, T]:
 def intersect(*lists: Iterable[T]) -> list[T]:
   return list(set.intersection(*map(set, lists)))
 
-def ensure_tuple(value: Union[T, tuple[T, ...]]) -> tuple[T, ...]:
+def ensure_tuple(value: T | tuple[T, ...]) -> tuple[T, ...]:
   return value if isinstance(value, tuple) else (value,)
 
 def key_of(dicts: Iterable[dict[T, U]], key: T) -> list[U]:
@@ -153,6 +155,7 @@ def get_at(d: dict, keys: Iterable[Any], default: T) -> T:
   return cast(Any, d)
 
 def sized_partitions(values: Iterable[T], part_size: int) -> list[list[T]]:
+  # "chunk"
   if not isinstance(values, list):
     values = list(values)
   num_parts = (len(values) / part_size).__ceil__()
@@ -169,11 +172,10 @@ def _cat_tile(cats, n_tile):
 
 def df_from_array(
   value_cols: dict[str, np.ndarray],
-  dim_labels: list[tuple[str, list[Union[str, int, float]]]],
+  dim_labels: list[tuple[str, list[str | int | float]]],
   indexed=False,
 ):
   import pandas as pd
-  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())
   array_offsets = [
@@ -182,12 +184,12 @@ def df_from_array(
   ]
   category_cols = {
     dim: _cat_tile(pd.Categorical(labels).repeat(repeats), tiles)
-    for dim, labels, (repeats, tiles) in zip(dim_names, dim_labels, array_offsets)
+    for (dim, labels), (repeats, tiles) in zip(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:
-    df = df.set_index(dim_names)
+    df = df.set_index([name for name, _ in dim_labels])
   return df
 
 StrFilter = Callable[[str], bool]

relib-1.0.9.dist-info/METADATA

@@ -0,0 +1,16 @@
+Metadata-Version: 2.3
+Name: relib
+Version: 1.0.9
+Project-URL: Repository, https://github.com/Reddan/relib.git
+Author: Hampus Hallman
+License: Copyright 2023 Hampus Hallman
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+License-File: LICENSE
+Requires-Python: >=3.12
+Requires-Dist: numpy>=2.1.0
+Requires-Dist: termcolor>=2.4.0

relib-1.0.9.dist-info/RECORD

@@ -0,0 +1,9 @@
+relib/__init__.py,sha256=nz5c2tIomUVXKzQshrt4s9dS_HirItWNa9Aem-9q368,483
+relib/hashing.py,sha256=6iAPRiJI_4jaSooZRFJnqK2limXqTmErzcwpd050LAA,8943
+relib/measure_duration.py,sha256=mTFvqGxKN2vTuHXEaWGHqZ-zm68Gbynxt1u6BHzKEQ8,511
+relib/raypipe.py,sha256=ynEoXs1dnD-360_uQC8v89xjiilt3knpocXpFaQ3plA,1905
+relib/utils.py,sha256=h_5UMOuXG4u6haslL88ESPKIBfBFhUvdobzo3eIrQs8,6256
+relib-1.0.9.dist-info/METADATA,sha256=nhtMQJqUijL1z2EQWwrP46gRNGFn7TWTdp1psXncw7w,1350
+relib-1.0.9.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+relib-1.0.9.dist-info/licenses/LICENSE,sha256=t9LfkVbmcvZjP0x3Sq-jR38UfTNbNtRQvc0Q8HWmLak,1054
+relib-1.0.9.dist-info/RECORD,,

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

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

relib-1.0.7.dist-info/METADATA

@@ -1,12 +0,0 @@
-Metadata-Version: 2.1
-Name: relib
-Version: 1.0.7
-Home-page: https://github.com/Reddan/relib
-Author: Hampus Hallman
-Author-email: me@hampushallman.com
-License: MIT
-Requires-Python: ~=3.10
-License-File: LICENSE.txt
-Requires-Dist: termcolor
-Requires-Dist: numpy
-

relib-1.0.7.dist-info/RECORD

@@ -1,10 +0,0 @@
-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.7.dist-info/top_level.txt

@@ -1 +0,0 @@
-relib