imt-ring 1.6.5__py3-none-any.whl → 1.6.7__py3-none-any.whl

{imt_ring-1.6.5.dist-info → imt_ring-1.6.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imt-ring
-Version: 1.6.5
+Version: 1.6.7
 Summary: RING: Recurrent Inertial Graph-based Estimator
 Author-email: Simon Bachhuber <simon.bachhuber@fau.de>
 Project-URL: Homepage, https://github.com/SimiPixel/ring
@@ -16,16 +16,11 @@ Requires-Dist: jaxopt
 Requires-Dist: numpy
 Requires-Dist: flax
 Requires-Dist: tqdm
-Requires-Dist: wget
-Requires-Dist: h5py
 Requires-Dist: imt-tree-utils
 Requires-Dist: wandb
 Requires-Dist: optax
 Requires-Dist: dm-haiku
-Requires-Dist: joblib
 Requires-Dist: pyyaml
-Requires-Dist: qmt
-Requires-Dist: scipy
 Provides-Extra: dev
 Requires-Dist: mkdocs ; extra == 'dev'
 Requires-Dist: mkdocs-material ; extra == 'dev'

{imt_ring-1.6.5.dist-info → imt_ring-1.6.7.dist-info}/RECORD

@@ -14,7 +14,7 @@ ring/algorithms/custom_joints/rr_imp_joint.py,sha256=_YJK0p8_0MHFtr1NuGnNZoxTbwa
 ring/algorithms/custom_joints/rr_joint.py,sha256=jnRtjtOCALMaq2_0bcu2d7qgfQ6etXpoh43MioRaDmY,1000
 ring/algorithms/custom_joints/suntay.py,sha256=tOEGM304XciHO4pmvxr4faA4xXVO4N2HlPdFmXKbcrw,16726
 ring/algorithms/generator/__init__.py,sha256=bF-CW3x2x-o6KWESKy-DuxzZPh3UNSjJb_MaAcSHGsQ,277
-ring/algorithms/generator/base.py,sha256=00789VeBrYcKx5BjsiGu-d4UzO6FGRz-YUKiiLUOL2Q,14497
+ring/algorithms/generator/base.py,sha256=YKUr9UnjFbWAYjP8V1j0FaKGVpK92W6E1iuoet7qBNg,14522
 ring/algorithms/generator/batch.py,sha256=ylootnXmj-JyuB_f5OCknHst9wFKO3gkjQbMrFNXY2g,2513
 ring/algorithms/generator/finalize_fns.py,sha256=LUw1Wc2YrmMRRh4RF704ob3bZOXktAZAbbLoBm_p1yw,9131
 ring/algorithms/generator/motion_artifacts.py,sha256=2VJbldVDbI3PSyboshIbtYvSAKzBBwGV7cQfYjqvluM,9167
@@ -72,18 +72,18 @@ ring/sys_composer/__init__.py,sha256=5J_JJJIHfTPcpxh0v4FqiOs81V1REPUd7pgiw2nAN5E
 ring/sys_composer/delete_sys.py,sha256=cIM9KbyLfg7B9121g7yjzuFbjeNu9cil1dPavAYEgzk,3408
 ring/sys_composer/inject_sys.py,sha256=Mj-q-mUjXKwkg-ol6IQAjf9IJfk7pGhez0_WoTKTgm0,3503
 ring/sys_composer/morph_sys.py,sha256=2GpPtS5hT0eZMptdGpt30Hc97OykJNE67lEVRf7sHrc,12700
-ring/utils/__init__.py,sha256=M9bR1-SYtmF9c4mTRIrGuIQws3K2aKUQxbpltIDkgZQ,739
+ring/utils/__init__.py,sha256=MHHavc8YfjBlmB-zAV42QEQS_ebW7cy0lhWXEVyQU7s,720
 ring/utils/backend.py,sha256=cKSi9sB59texqKzNVASTDczGKLCBL8VVDiP7TNdj41k,1294
 ring/utils/batchsize.py,sha256=FbOii7MDP4oPZd9GJOKehFatfnb6WZ0b9z349iZYs1A,1786
 ring/utils/colab.py,sha256=ZLHwP0jNQUsmZJU4l68a5djULPi6T-jYNNHevjIoMn8,1631
 ring/utils/hdf5.py,sha256=BzXwVypZmEZeHVgeGZ78YYdi10NEQtnPhdrb8dQAXo0,5856
 ring/utils/normalizer.py,sha256=67L2BU1MRsMT4pD41ta3JJMppLN0ozFmnwrmXDtnqrQ,1698
-ring/utils/path.py,sha256=hAfSlqRi-ew536RnjDDM7IKapdMJc-EvhrR0Y-BCFWc,1265
+ring/utils/path.py,sha256=zRPfxYNesvgefkddd26oar6f9433LkMGkhp9dF3rPUs,1926
 ring/utils/randomize_sys.py,sha256=G_vBIo0OwQkXL2u0djwbaoaeb02C4LQCTNNloOYIU2M,3699
-ring/utils/utils.py,sha256=U7CY04ehx8pZkUn7clSDBcTWc89fiTJEjyOxIVbz4GM,6529
+ring/utils/utils.py,sha256=oGC7kh19s5zvmnUvWy8B3fBl9loVU58ppz91osk2m3w,6550
 ring/utils/register_gym_envs/__init__.py,sha256=PtPIRBQJ16339xZ9G9VpvqrvcGbQ_Pk_SUz4tQPa9nQ,94
 ring/utils/register_gym_envs/saddle.py,sha256=tA5CyW_akSXyDm0xJ83CtOrUMVElH0f9vZtEDDJQalI,4422
-imt_ring-1.6.5.dist-info/METADATA,sha256=-uLPyzjqj0rkrFf9xa_fmB0qT7mxwItukQQaypelW-4,3922
-imt_ring-1.6.5.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
-imt_ring-1.6.5.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
-imt_ring-1.6.5.dist-info/RECORD,,
+imt_ring-1.6.7.dist-info/METADATA,sha256=qmP76hCRm-7QsTvxE_D4D36JZoCzVhhrsiujBvdL5Vw,3820
+imt_ring-1.6.7.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+imt_ring-1.6.7.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
+imt_ring-1.6.7.dist-info/RECORD,,

{imt_ring-1.6.5.dist-info → imt_ring-1.6.7.dist-info}/WHEEL

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

ring/algorithms/generator/base.py

@@ -322,7 +322,7 @@ def _build_mconfig_batched_generator(
         if add_y_relpose:
             pipe.append(finalize_fns.RelPose(sys_noimu))
         if add_y_rootincl:
-            pipe.append(finalize_fns.RootIncl(sys_noimu))
+            pipe.append(finalize_fns.RootIncl(sys_noimu, **add_y_rootincl_kwargs))
         if use_link_number_in_Xy:
             pipe.append(finalize_fns.Names2Indices(sys_noimu))
 

ring/utils/__init__.py

@@ -1,4 +1,3 @@
-from . import hdf5
 from . import randomize_sys
 from .batchsize import batchsize_thresholds
 from .batchsize import distribute_batchsize

ring/utils/path.py

@@ -12,6 +12,24 @@ def parse_path(
     mkdir: bool = True,
     require_is_file: bool = False,
 ) -> str:
+    """Utility for performing actions on a path
+
+    Args:
+        path (str): The (root) path to file or folder
+        extension (Optional[str], optional): The file extension. Defaults to None.
+        file_exists_ok (bool, optional): If false, will error if file already exists.
+            Defaults to True.
+        mkdir (bool, optional): Will create all subdirs in the process.
+            Defaults to True.
+        require_is_file (bool, optional): if true, will error if file does not
+            already exist. Defaults to False.
+
+    Raises:
+        Exception: _description_
+
+    Returns:
+        str: The home-expanded path to folder or file
+    """
     path = Path(os.path.expanduser(path))
 
     for p in join_paths:

ring/utils/utils.py

@@ -115,7 +115,7 @@ def pytree_deepcopy(tree):
     "Recursivley copies a pytree."
     if isinstance(tree, (int, float, jax.Array)):
         return tree
-    elif isinstance(tree, np.ndarray):
+    elif isinstance(tree, np.ndarray) or np.isscalar(tree):
         return tree.copy()
     elif isinstance(tree, list):
         return [pytree_deepcopy(ele) for ele in tree]