imt-ring 1.6.18__py3-none-any.whl → 1.6.20__py3-none-any.whl

{imt_ring-1.6.18.dist-info → imt_ring-1.6.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: imt-ring
-Version: 1.6.18
+Version: 1.6.20
 Summary: RING: Recurrent Inertial Graph-based Estimator
 Author-email: Simon Bachhuber <simon.bachhuber@fau.de>
 Project-URL: Homepage, https://github.com/SimiPixel/ring
@@ -98,6 +98,14 @@ import os
 os.environ["MUJOCO_GL"] = "egl"
 ```
 
+#### Windows-related: ImportError: DLL load failed while importing ...
+
+> ImportError: DLL load failed while importing _multiarray_umath: Das angegebene Modul wurde nicht gefunden.
+
+Solution:
+1. `pip uninstall -y jax jaxlib`
+2. `conda install -c conda-forge jax`
+
 ## Publications
 
 The following publications utilize this software library, and refer to it as the *Random Chain Motion Generator (RCMG)* (more specifically the function `ring.RCMG`):

{imt_ring-1.6.18.dist-info → imt_ring-1.6.20.dist-info}/RECORD

@@ -54,8 +54,8 @@ ring/io/xml/to_xml.py,sha256=fohb-jWMf2cxVdT5dmknsGyrNMseICSbKEz_urbaWbQ,3407
 ring/ml/__init__.py,sha256=nbh48gaswWeY4S4vT1sply_3ROj2DQ7agjoLR4Ho3T8,1517
 ring/ml/base.py,sha256=lfwEZLBDglOSRWChUHoH1kezefhttPV9TMEpNIqsMNw,9972
 ring/ml/callbacks.py,sha256=W19QF6_uvaNCjs8ObsjNXD7mv9gFgJBixdRSbB_BynE,13301
-ring/ml/ml_utils.py,sha256=1GXJfeoXbwCbRdYA2np3CbJpSupaw4eyf3quh9y4BO0,6462
-ring/ml/optimizer.py,sha256=fWyF__ezUltrA16SLfOC1jvS3zBh9NJsMYa6-V0frhs,4709
+ring/ml/ml_utils.py,sha256=xqy9BnLy8IKVqkFS9mlZsGJXSbThI9zZxZ5rhl8LSI8,7144
+ring/ml/optimizer.py,sha256=TZF0_LmnewzmGVso-zIQJtpWguUW0fW3HeRpIdG_qoI,4763
 ring/ml/ringnet.py,sha256=mef7jyN2QcApJmQGH3HYZyTV-00q8YpsYOKhW0-ku1k,8973
 ring/ml/rnno_v1.py,sha256=2qE08OIvTJ5PvSxKpYGzGSrvEImWrdAT_qslZ7jP5tA,1372
 ring/ml/train.py,sha256=huUfMK6eotS6BRrQKoZ-AUG0um3jlqpfQFZNJT8LKiE,10854
@@ -86,7 +86,7 @@ ring/utils/randomize_sys.py,sha256=G_vBIo0OwQkXL2u0djwbaoaeb02C4LQCTNNloOYIU2M,3
 ring/utils/utils.py,sha256=tJaWXLGOTwkxJQj2l23dX97wO3aZYhM2qd7eNuMRs84,6907
 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.18.dist-info/METADATA,sha256=BhREl-3Q3LDO-ugNEfitQuW2ZDCR-ng-Hleb9FVG6Ps,3833
-imt_ring-1.6.18.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-imt_ring-1.6.18.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
-imt_ring-1.6.18.dist-info/RECORD,,
+imt_ring-1.6.20.dist-info/METADATA,sha256=uOgzTpTz7MgDEIGhL7onaD_r64DlL6ALSnaT5t8u3HY,4097
+imt_ring-1.6.20.dist-info/WHEEL,sha256=a7TGlA-5DaHMRrarXjVbQagU3Man_dCnGIWMJr5kRWo,91
+imt_ring-1.6.20.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
+imt_ring-1.6.20.dist-info/RECORD,,

{imt_ring-1.6.18.dist-info → imt_ring-1.6.20.dist-info}/WHEEL

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

ring/ml/ml_utils.py

@@ -3,6 +3,7 @@ from functools import partial
 import os
 from pathlib import Path
 import pickle
+import shutil
 import time
 from typing import Optional, Protocol
 import warnings
@@ -107,7 +108,7 @@ class WandbLogger(MixinLogger):
         wandb.log(data)
 
     def log_params(self, path: str):
-        wandb.save(path, policy="now")
+        self.wandb_save(path)
 
     def log_video(
         self,
@@ -117,7 +118,7 @@ class WandbLogger(MixinLogger):
         step: Optional[int] = None,
     ):
         # TODO >>>
-        wandb.save(path, policy="now")
+        self.wandb_save(path)
         return
         # <<<
         data = {"video": wandb.Video(path, caption=caption, fps=fps)}
@@ -127,10 +128,10 @@ class WandbLogger(MixinLogger):
 
     def log_image(self, path: str, caption: Optional[str] = None):
         # wandb.log({"image": wandb.Image(path, caption=caption)})
-        wandb.save(path, policy="now")
+        self.wandb_save(path)
 
     def log_txt(self, path: str, wait: bool = True):
-        wandb.save(path, policy="now")
+        self.wandb_save(path)
         # TODO: `wandb` is not async at all?
         if wait:
             time.sleep(3)
@@ -138,6 +139,23 @@ class WandbLogger(MixinLogger):
     def close(self):
         wandb.run.finish()
 
+    @staticmethod
+    def wandb_save(path):
+        if wandb.run is not None and wandb.run.settings._offline:
+            # Create a dedicated directory in the WandB run directory to store copies
+            # of files
+            destination_dir = os.path.join(wandb.run.dir, "copied_files")
+            os.makedirs(destination_dir, exist_ok=True)
+
+            # Copy the file to this new location
+            copied_file_path = os.path.join(destination_dir, os.path.basename(path))
+            shutil.copy2(path, copied_file_path)
+
+            # Use wandb.save to save the copied file (now a true copy)
+            wandb.save(copied_file_path)
+        else:
+            wandb.save(path, policy="now")
+
 
 def _flatten_convert_filter_nested_dict(
     metrices: NestedDict, filter_nan_inf: bool = True

ring/ml/optimizer.py

@@ -48,7 +48,11 @@ def make_optimizer(
             if adap_clip is not None
             else optax.identity()
         ),
-        optax.clip_by_global_norm(0.2) if glob_clip is not None else optax.identity(),
+        (
+            optax.clip_by_global_norm(glob_clip)
+            if glob_clip is not None
+            else optax.identity()
+        ),
         inner_opt(schedule, **inner_opt_kwargs),
     )
     optimizer = skip_large_update(