imt-ring 1.6.4__py3-none-any.whl → 1.6.6__py3-none-any.whl

{imt_ring-1.6.4.dist-info → imt_ring-1.6.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imt-ring
-Version: 1.6.4
+Version: 1.6.6
 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.6.4.dist-info → imt_ring-1.6.6.dist-info}/RECORD

@@ -80,10 +80,10 @@ 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/randomize_sys.py,sha256=G_vBIo0OwQkXL2u0djwbaoaeb02C4LQCTNNloOYIU2M,3699
-ring/utils/utils.py,sha256=k7t-QxMWrNRnjfNB9rSobmLCmhJigE8__gkT-Il0Ee4,6492
+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.4.dist-info/METADATA,sha256=3uiX-NEHrXZ4QurQPYQUPRlul7aVLBZYIr0w17Yo54E,3922
-imt_ring-1.6.4.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
-imt_ring-1.6.4.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
-imt_ring-1.6.4.dist-info/RECORD,,
+imt_ring-1.6.6.dist-info/METADATA,sha256=izx_ZcizbG0DLoo7R2zzAbt-o-Zheh1hxPd7xTnxznY,3922
+imt_ring-1.6.6.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
+imt_ring-1.6.6.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
+imt_ring-1.6.6.dist-info/RECORD,,

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]
@@ -210,8 +210,9 @@ def replace_elements_w_nans(
         if _is_nan(ele, i):
             while True:
                 j = random.choice(include_elements)
-                if not _is_nan(list_of_data[j], j):
-                    ele = list_of_data[j]
+                ele_j = list_of_data[j]
+                if not _is_nan(ele_j, j):
+                    ele = pytree_deepcopy(ele_j)
                     break
         list_of_data_nonan.append(ele)
     return list_of_data_nonan