imt-ring 1.3.8__py3-none-any.whl → 1.3.9__py3-none-any.whl

{imt_ring-1.3.8.dist-info → imt_ring-1.3.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imt-ring
-Version: 1.3.8
+Version: 1.3.9
 Summary: RING: Recurrent Inertial Graph-based Estimator
 Author-email: Simon Bachhuber <simon.bachhuber@fau.de>
 Project-URL: Homepage, https://github.com/SimiPixel/ring

{imt_ring-1.3.8.dist-info → imt_ring-1.3.9.dist-info}/RECORD

@@ -1,11 +1,11 @@
 ring/__init__.py,sha256=iNvbAZi7Qfa69IbL1z4lB7zHL8WusV5fBrKah2la-Gc,1566
 ring/algebra.py,sha256=F0GwbP8LQP5qGVkoMUYJmkp9Hn2nKAVIkCVYDEjNjGU,3128
-ring/base.py,sha256=99GuspRH4QtRRJTAgyvS02FFxoaBptSsz_GPczX8vw0,33947
+ring/base.py,sha256=YFPrUWelWswEhq8x8Byv-5pK64mipiGW6x5IlMr4we4,33803
 ring/maths.py,sha256=jJr_kr78-XDce8B4tXQ2Li-jBntVQhaS8csxglCsj8A,12193
 ring/spatial.py,sha256=nmZ-UhRanhyM34bez8uCS4wMwaKqLkuEbgKGP5XNH60,2351
 ring/algorithms/__init__.py,sha256=t3YXcgqMJxadUjFiILVD0HlQRPLdrQyc8aKiB36w0vE,1701
 ring/algorithms/_random.py,sha256=M9JQSMXSUARWuzlRLP3Wmkuntrk9LZpP30p4_IPgDB4,13805
-ring/algorithms/dynamics.py,sha256=nqq5I0RYSbHNlGiLMlohz08IfL9Njsrid4upDnwkGbI,10629
+ring/algorithms/dynamics.py,sha256=_TwclBXe6vi5C5iJWAIeUIJEIMHQ_1QTmnHvCEpVO0M,10867
 ring/algorithms/jcalc.py,sha256=6olMYQtgKZE5KBEAHF0Rqxe__1wcZQVEiLgm1vO7_Gw,28260
 ring/algorithms/kinematics.py,sha256=DOboHI517Vx0pRJUFZtZPmK_qFaiKiQe-37B-M0aC-c,7422
 ring/algorithms/sensors.py,sha256=MICO9Sn0AfoqRx_9KWR3hufsIID-K6SOIg3oPDgsYMU,17869
@@ -14,10 +14,10 @@ ring/algorithms/custom_joints/rr_imp_joint.py,sha256=a3JT0w7pB94kZ95eBR8ZO853eSe
 ring/algorithms/custom_joints/rr_joint.py,sha256=jnRtjtOCALMaq2_0bcu2d7qgfQ6etXpoh43MioRaDmY,1000
 ring/algorithms/custom_joints/suntay.py,sha256=7-kym1kMDwqYD_2um1roGcBeB8BlTCPe1wljuNGNARA,16676
 ring/algorithms/generator/__init__.py,sha256=p4ucl0zQtp5NwNoXIRjmTzGGRu2WOAWFfNmYRPwQles,912
-ring/algorithms/generator/base.py,sha256=AKH7GXEmRGV1kK8okiqa12uq0Ah9VYlqgdLw-99oFoQ,14840
-ring/algorithms/generator/batch.py,sha256=ABmaNguQKVStphHN9h5_oNaNC2Un_YoR0iswdKaLOHs,9178
-ring/algorithms/generator/motion_artifacts.py,sha256=Hk088xPYxnKsqmQEtr_bqRWq5zYBVM7PEG8IvHmMY6g,8711
-ring/algorithms/generator/pd_control.py,sha256=3pOaYig26vmp8gippDfy2KNJRZO3kr0rGd_PBIuEROM,5759
+ring/algorithms/generator/base.py,sha256=sr-YZkjd8pZJAI5vFG_IqOO4AEeiEYtXr8uUsPMS6Q4,14779
+ring/algorithms/generator/batch.py,sha256=kNlq78W-nAtbp6Xe82UjbPY-rXX2alGLxTokTITSbAc,9226
+ring/algorithms/generator/motion_artifacts.py,sha256=_kiAl1VHoX1fW5AUlXOtPBWyHIIFof_M78AP-m9f1ME,8790
+ring/algorithms/generator/pd_control.py,sha256=XJ_Gd5AkIRh-jBrMfQyMXjVwhx2gCNHznjzFbmAwhZs,5767
 ring/algorithms/generator/randomize.py,sha256=G_vBIo0OwQkXL2u0djwbaoaeb02C4LQCTNNloOYIU2M,3699
 ring/algorithms/generator/transforms.py,sha256=nvNDvM20tEw9Zd0ra0TxA25uf01L40Y2UKvtQmOrlGo,12782
 ring/algorithms/generator/types.py,sha256=CAhvDK5qiHnrGtkCVccB07doiz_D6lHJ35B7sW0pyZA,1110
@@ -56,7 +56,7 @@ ring/ml/callbacks.py,sha256=W19QF6_uvaNCjs8ObsjNXD7mv9gFgJBixdRSbB_BynE,13301
 ring/ml/ml_utils.py,sha256=hQEmeZoahdJyFrz0NZXYi1Yijl7GvPBdqwzZBzlUIUM,7638
 ring/ml/optimizer.py,sha256=fWyF__ezUltrA16SLfOC1jvS3zBh9NJsMYa6-V0frhs,4709
 ring/ml/ringnet.py,sha256=rgje5AKUKpT8K-vbE9_SgZ3IijR8TJEHnaqxsE57Mhc,8617
-ring/ml/rnno_v1.py,sha256=MpuDyd9B32Qq_gRqiZf4x1q0-2HzADK5JhjjzMWQfkE,1026
+ring/ml/rnno_v1.py,sha256=T4SKG7iypqn2HBQLKhDmJ2Slj2Z5jtUBHvX_6aL8pyM,1103
 ring/ml/train.py,sha256=huUfMK6eotS6BRrQKoZ-AUG0um3jlqpfQFZNJT8LKiE,10854
 ring/ml/training_loop.py,sha256=CEokvPQuuk_WCd-J60ZDodJYcPVvyxLfgXDr_DnbzRI,3359
 ring/ml/params/0x13e3518065c21cd8.pickle,sha256=Zh2k1zK-TNxJl5F7nyTeQ9001qqRE_dfvaq1HWV287A,9355838
@@ -78,7 +78,7 @@ 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/utils.py,sha256=mIcKNv5v2de8HrG7bAhl2bNfmwkMZyIIwFkJq2XWMOI,5357
-imt_ring-1.3.8.dist-info/METADATA,sha256=V0FWiPAeTUFF7hD-Y4cucpF9HGZETgXgWaZgYIz41ho,3104
-imt_ring-1.3.8.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-imt_ring-1.3.8.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
-imt_ring-1.3.8.dist-info/RECORD,,
+imt_ring-1.3.9.dist-info/METADATA,sha256=H65-QICwM4mtRPumYJbrenN74nmiMBGbeV3pecKEeOg,3104
+imt_ring-1.3.9.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+imt_ring-1.3.9.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
+imt_ring-1.3.9.dist-info/RECORD,,

ring/algorithms/dynamics.py

@@ -1,7 +1,9 @@
 from typing import Optional, Tuple
+import warnings
 
 import jax
 import jax.numpy as jnp
+
 from ring import algebra
 from ring import base
 from ring import maths
@@ -213,7 +215,7 @@ def forward_dynamics(
     q: jax.Array,
     qd: jax.Array,
     tau: jax.Array,
-    mass_mat_inv: jax.Array,
+    # mass_mat_inv: jax.Array,
 ) -> Tuple[jax.Array, jax.Array]:
     C = inverse_dynamics(sys, qd, jnp.zeros_like(qd))
     mass_matrix = compute_mass_matrix(sys)
@@ -235,6 +237,11 @@ def forward_dynamics(
 
         mass_mat_inv = jax.scipy.linalg.solve(mass_matrix, eye, assume_a="pos")
     else:
+        warnings.warn(
+            f"You are using `sys.mass_mat_iters`={sys.mass_mat_iters} which is >0. "
+            "This feature is currently not fully supported. See the local TODO."
+        )
+        mass_mat_inv = jnp.diag(jnp.ones((sys.qd_size(),)))
         mass_mat_inv = _inv_approximate(mass_matrix, mass_mat_inv, sys.mass_mat_iters)
 
     return mass_mat_inv @ qf_smooth, mass_mat_inv
@@ -254,9 +261,8 @@ def _strapdown_integration(
 def _semi_implicit_euler_integration(
     sys: base.System, state: base.State, taus: jax.Array
 ) -> base.State:
-    qdd, mass_mat_inv = forward_dynamics(
-        sys, state.q, state.qd, taus, state.mass_mat_inv
-    )
+    qdd, mass_mat_inv = forward_dynamics(sys, state.q, state.qd, taus)
+    del mass_mat_inv
     qd_next = state.qd + sys.dt * qdd
 
     q_next = []
@@ -277,7 +283,7 @@ def _semi_implicit_euler_integration(
     sys.scan(q_integrate, "qdl", state.q, qd_next, sys.link_types)
     q_next = jnp.concatenate(q_next)
 
-    state = state.replace(q=q_next, qd=qd_next, mass_mat_inv=mass_mat_inv)
+    state = state.replace(q=q_next, qd=qd_next)
     return state
 
 

ring/algorithms/generator/base.py

@@ -4,6 +4,7 @@ import warnings
 
 import jax
 import jax.numpy as jnp
+import tqdm
 import tree_utils
 
 from ring import base
@@ -83,10 +84,14 @@ class RCMG:
             ), "If `randomize_anchors`, then only one system is expected"
             sys = randomize.randomize_anchors(sys[0], **randomize_anchors_kwargs)
 
-        zip_sys_config = False
         if randomize_hz:
-            zip_sys_config = True
             sys, config = randomize.randomize_hz(sys, config, **randomize_hz_kwargs)
+        else:
+            # create zip
+            N_sys = len(sys)
+            sys = sum([len(config) * [s] for s in sys], start=[])
+            config = N_sys * config
+            assert len(sys) == len(config)
 
         if sys_ml is None:
             # TODO
@@ -97,17 +102,10 @@ class RCMG:
             sys_ml = sys[0]
 
         self.gens = []
-        if zip_sys_config:
-            for _sys, _config in zip(sys, config):
-                self.gens.append(
-                    partial_build_gen(sys=_sys, config=_config, sys_ml=sys_ml)
-                )
-        else:
-            for _sys in sys:
-                for _config in config:
-                    self.gens.append(
-                        partial_build_gen(sys=_sys, config=_config, sys_ml=sys_ml)
-                    )
+        for _sys, _config in tqdm.tqdm(
+            zip(sys, config), desc="building generators", total=len(sys)
+        ):
+            self.gens.append(partial_build_gen(sys=_sys, config=_config, sys_ml=sys_ml))
 
     def _to_data(self, sizes, seed):
         return batch.batch_generators_eager_to_list(self.gens, sizes, seed=seed)

ring/algorithms/generator/batch.py

@@ -86,7 +86,11 @@ def batch_generators_eager_to_list(
 
     key = jax.random.PRNGKey(seed)
     data = []
-    for gen, size in tqdm(zip(generators, sizes), desc="eager data generation"):
+    for gen, size in tqdm(
+        zip(generators, sizes),
+        desc="executing generators",
+        total=len(sizes),
+    ):
 
         n_calls = _number_of_executions_required(size)
         # decrease size by n_calls times

ring/algorithms/generator/motion_artifacts.py

@@ -49,6 +49,7 @@ def inject_subsystems(
     rotational_damp: float = 0.1,
     translational_stif: float = 50.0,
     translational_damp: float = 0.1,
+    disable_warning: bool = False,
     **kwargs,
 ) -> base.System:
     imu_idx_to_name_map = {sys.name_to_idx(imu): imu for imu in sys.findall_imus()}
@@ -92,10 +93,11 @@ def inject_subsystems(
     # TODO set all joint_params to zeros; they can not be preserved anyways and
     # otherwise many warnings will be rose
     # instead warn explicitly once now and move on
-    warnings.warn(
-        "`sys.links.joint_params` has been set to zero, this might lead to "
-        "unexpected behaviour unless you use `randomize_joint_params`"
-    )
+    if not disable_warning:
+        warnings.warn(
+            "`sys.links.joint_params` has been set to zero, this might lead to "
+            "unexpected behaviour unless you use `randomize_joint_params`"
+        )
     joint_params_zeros = tree_utils.tree_zeros_like(sys.links.joint_params)
     sys = sys.replace(links=sys.links.replace(joint_params=joint_params_zeros))
 

ring/algorithms/generator/pd_control.py

@@ -4,6 +4,7 @@ from typing import Optional
 from flax import struct
 import jax
 import jax.numpy as jnp
+
 from ring import base
 from ring.algorithms import dynamics
 from ring.algorithms import jcalc
@@ -49,7 +50,7 @@ def _pd_control(P: jax.Array, D: Optional[jax.Array] = None):
         assert sys.q_size() == q_ref.shape[1], f"q_ref.shape = {q_ref.shape}"
         assert sys.qd_size() == P.size
         if D is not None:
-            sys.qd_size() == D.size
+            assert sys.qd_size() == D.size
 
         q_ref_as_dict = {}
         qd_ref_as_dict = {}

ring/base.py

@@ -997,13 +997,11 @@ class State(_Base):
         q (jax.Array): System state in minimal coordinates (equals `sys.q_size()`)
         qd (jax.Array): System velocity in minimal coordinates (equals `sys.qd_size()`)
         x: (Transform): Maximal coordinates of all links. From epsilon-to-link.
-        mass_mat_inv (jax.Array): Inverse of the mass matrix. Internal usage.
     """
 
     q: jax.Array
     qd: jax.Array
     x: Transform
-    mass_mat_inv: jax.Array
 
     @classmethod
     def create(
@@ -1057,4 +1055,4 @@ class State(_Base):
         if x is None:
             x = Transform.zero((sys.num_links(),))
 
-        return cls(q, qd, x, jnp.diag(jnp.ones((sys.qd_size(),))))
+        return cls(q, qd, x)

ring/ml/rnno_v1.py

@@ -1,4 +1,4 @@
-from typing import Sequence
+from typing import Optional, Sequence
 
 import haiku as hk
 import jax
@@ -12,14 +12,18 @@ def rnno_v1_forward_factory(
     layernorm: bool = True,
     act_fn_linear=jax.nn.relu,
     act_fn_rnn=jax.nn.elu,
+    lam: Optional[tuple[int]] = None,
 ):
+    # unused
+    del lam
+
     @hk.without_apply_rng
     @hk.transform_with_state
     def forward_fn(X):
         assert X.shape[-2] == 1
 
         for i, n_units in enumerate(rnn_layers):
-            state = hk.get_state(f"rnn_{i}", shape=[n_units], init=jnp.zeros)
+            state = hk.get_state(f"rnn_{i}", shape=[1, n_units], init=jnp.zeros)
             X, state = hk.dynamic_unroll(hk.GRU(n_units), X, state)
             hk.set_state(f"rnn_{i}", state)