imt-ring 1.6.15__tar.gz → 1.6.16__tar.gz

{imt_ring-1.6.15 → imt_ring-1.6.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imt-ring
-Version: 1.6.15
+Version: 1.6.16
 Summary: RING: Recurrent Inertial Graph-based Estimator
 Author-email: Simon Bachhuber <simon.bachhuber@fau.de>
 Project-URL: Homepage, https://github.com/SimiPixel/ring
@@ -32,11 +32,11 @@ Requires-Dist: pytest-xdist; extra == "dev"
 Requires-Dist: nbmake; extra == "dev"
 
 <p align="center">
-<img src="https://raw.githubusercontent.com/SimiPixel/ring/main/docs/img/icon.svg" height="200" />
+<img src="https://raw.githubusercontent.com/simon-bachhuber/ring/main/docs/img/icon.svg" height="200" />
 </p>
 
 # Recurrent Inertial Graph-based Estimator (RING)
-<img src="https://raw.githubusercontent.com/SimiPixel/ring/main/docs/img/coverage_badge.svg" height="20" />
+<img src="https://raw.githubusercontent.com/simon-bachhuber/ring/main/docs/img/coverage_badge.svg" height="20" />
 
 ## Installation
 

{imt_ring-1.6.15 → imt_ring-1.6.16}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "imt-ring"
-version = "1.6.15"
+version = "1.6.16"
 authors = [
   { name="Simon Bachhuber", email="simon.bachhuber@fau.de" },
 ]

{imt_ring-1.6.15 → imt_ring-1.6.16}/readme.md

@@ -1,9 +1,9 @@
 <p align="center">
-<img src="https://raw.githubusercontent.com/SimiPixel/ring/main/docs/img/icon.svg" height="200" />
+<img src="https://raw.githubusercontent.com/simon-bachhuber/ring/main/docs/img/icon.svg" height="200" />
 </p>
 
 # Recurrent Inertial Graph-based Estimator (RING)
-<img src="https://raw.githubusercontent.com/SimiPixel/ring/main/docs/img/coverage_badge.svg" height="20" />
+<img src="https://raw.githubusercontent.com/simon-bachhuber/ring/main/docs/img/coverage_badge.svg" height="20" />
 
 ## Installation
 

{imt_ring-1.6.15 → imt_ring-1.6.16}/src/imt_ring.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imt-ring
-Version: 1.6.15
+Version: 1.6.16
 Summary: RING: Recurrent Inertial Graph-based Estimator
 Author-email: Simon Bachhuber <simon.bachhuber@fau.de>
 Project-URL: Homepage, https://github.com/SimiPixel/ring
@@ -32,11 +32,11 @@ Requires-Dist: pytest-xdist; extra == "dev"
 Requires-Dist: nbmake; extra == "dev"
 
 <p align="center">
-<img src="https://raw.githubusercontent.com/SimiPixel/ring/main/docs/img/icon.svg" height="200" />
+<img src="https://raw.githubusercontent.com/simon-bachhuber/ring/main/docs/img/icon.svg" height="200" />
 </p>
 
 # Recurrent Inertial Graph-based Estimator (RING)
-<img src="https://raw.githubusercontent.com/SimiPixel/ring/main/docs/img/coverage_badge.svg" height="20" />
+<img src="https://raw.githubusercontent.com/simon-bachhuber/ring/main/docs/img/coverage_badge.svg" height="20" />
 
 ## Installation
 

{imt_ring-1.6.15 → imt_ring-1.6.16}/src/ring/algorithms/generator/finalize_fns.py

@@ -182,13 +182,16 @@ class DynamicalSimulation:
 
     @staticmethod
     def assert_test_system(sys: base.System) -> None:
-        "test that system has no zero mass bodies and no joints without damping"
+        "test that system has no zero mass leaf bodies and no joints without damping"
 
         def f(_, __, n, m, d):
-            assert d.size == 0 or m > 0, (
-                "Dynamic simulation is set to `True` which requires masses >= 0, "
-                f"but found body `{n}` with mass={float(m[0])}. This can lead to NaNs."
-            )
+            is_leaf_body = len(sys.children(n)) == 0
+            if is_leaf_body:
+                assert d.size == 0 or m > 0, (
+                    "Dynamic simulation is set to `True` which requires masses >= 0, "
+                    f"but found body `{n}` with mass={float(m[0])}. This can lead to "
+                    "NaNs."
+                )
 
             assert d.size == 0 or all(d > 0.0), (
                 "Dynamic simulation is set to `True` which requires dampings > 0, "

{imt_ring-1.6.15 → imt_ring-1.6.16}/src/ring/algorithms/jcalc.py

@@ -222,7 +222,9 @@ def _is_feasible_config1(c: MotionConfig) -> bool:
     def inside_box_checks(x_min, x_max, x0_min, x0_max) -> bool:
         return (x0_min >= x_min) and (x0_max <= x_max)
 
-    cond2 = inside_box_checks(c.pos_min, c.pos_max, c.pos0_min, c.pos0_max)
+    cond2 = inside_box_checks(
+        _to_float(c.pos_min, 0.0), _to_float(c.pos_max, 0.0), c.pos0_min, c.pos0_max
+    )
 
     return cond1 and cond2
 

{imt_ring-1.6.15 → imt_ring-1.6.16}/src/ring/ml/ringnet.py

@@ -87,6 +87,7 @@ def make_ring(
     link_output_normalize: bool = True,
     link_output_transform: Optional[Callable] = None,
     layernorm: bool = True,
+    layernorm_trainable: bool = True,
 ) -> SimpleNamespace:
 
     if link_output_normalize:
@@ -104,7 +105,11 @@ def make_ring(
         )
 
         inner_cell = StackedRNNCell(
-            celltype, hidden_state_dim, stack_rnn_cells, layernorm=layernorm
+            celltype,
+            hidden_state_dim,
+            stack_rnn_cells,
+            layernorm=layernorm,
+            layernorm_trainable=layernorm_trainable,
         )
         send_output = hk.nets.MLP([hidden_state_dim, link_output_dim])
         state = hk.get_state(
@@ -143,6 +148,7 @@ class StackedRNNCell(hk.Module):
         hidden_state_dim,
         stacks: int,
         layernorm: bool = False,
+        layernorm_trainable: bool = True,
         name: str | None = None,
     ):
         super().__init__(name)
@@ -150,6 +156,7 @@ class StackedRNNCell(hk.Module):
 
         self.cells = [cell(hidden_state_dim) for _ in range(stacks)]
         self.layernorm = layernorm
+        self.layernorm_trainable = layernorm_trainable
 
     def __call__(self, x, state):
         output = x
@@ -159,7 +166,9 @@ class StackedRNNCell(hk.Module):
             next_state.append(next_state_i)
 
             if self.layernorm:
-                output = hk.LayerNorm(-1, True, True)(output)
+                output = hk.LayerNorm(
+                    -1, self.layernorm_trainable, self.layernorm_trainable
+                )(output)
 
         return output, jnp.stack(next_state)
 

{imt_ring-1.6.15 → imt_ring-1.6.16}/tests/test_rcmg.py

@@ -37,6 +37,7 @@ def test_batch_generator(N: int, seed: int):
 def test_initial_ang_pos_values():
     T = 1.0
     bs = 8
+    pos_min, pos_max = -5.0, 5.0
     # system consists only of prismatic, and then revolute joint
     sys = ring.io.load_example("test_ang0_pos0")
 
@@ -48,6 +49,8 @@ def test_initial_ang_pos_values():
                 ang0_max=ang0_max,
                 pos0_min=pos0_min,
                 pos0_max=pos0_max,
+                pos_min=pos_min,
+                pos_max=pos_max,
                 T=T,
             ),
             finalize_fn=lambda key, q, x, sys: (q, x),
@@ -55,7 +58,7 @@ def test_initial_ang_pos_values():
         q, _ = gen(jax.random.PRNGKey(1))
         return q
 
-    for init_val in np.linspace(-5.0, 5.0, num=10):
+    for init_val in np.linspace(pos_min, pos_max, num=10):
         q = rcmg(init_val, init_val, init_val, init_val)
         np.testing.assert_allclose(q[:, 0, 0], init_val * jnp.ones((bs,)))
         np.testing.assert_allclose(q[:, 0, 1], wrap_to_pi(init_val * jnp.ones((bs,))))