imt-ring 1.6.5__tar.gz → 1.6.7__tar.gz

{imt_ring-1.6.5 → imt_ring-1.6.7}/PKG-INFO

@@ -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 → imt_ring-1.6.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "imt-ring"
-version = "1.6.5"
+version = "1.6.7"
 authors = [
   { name="Simon Bachhuber", email="simon.bachhuber@fau.de" },
 ]
@@ -23,16 +23,11 @@ dependencies = [
     "numpy",
     "flax",
     "tqdm",
-    "wget",
-    "h5py",
     "imt-tree-utils",
     "wandb",
     "optax",
     "dm-haiku",
-    "joblib",
     "pyyaml",
-    "qmt",
-    "scipy",
 ]
 
 [project.optional-dependencies]

{imt_ring-1.6.5 → imt_ring-1.6.7}/src/imt_ring.egg-info/PKG-INFO

@@ -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 → imt_ring-1.6.7}/src/imt_ring.egg-info/requires.txt

@@ -4,16 +4,11 @@ jaxopt
 numpy
 flax
 tqdm
-wget
-h5py
 imt-tree-utils
 wandb
 optax
 dm-haiku
-joblib
 pyyaml
-qmt
-scipy
 
 [dev]
 mkdocs

{imt_ring-1.6.5 → imt_ring-1.6.7}/src/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))
 

{imt_ring-1.6.5 → imt_ring-1.6.7}/src/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

{imt_ring-1.6.5 → imt_ring-1.6.7}/src/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:

{imt_ring-1.6.5 → imt_ring-1.6.7}/src/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]