imt-ring 1.6.18__tar.gz → 1.6.20__tar.gz

{imt_ring-1.6.18 → imt_ring-1.6.20}/PKG-INFO

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

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

{imt_ring-1.6.18 → imt_ring-1.6.20}/readme.md

@@ -65,6 +65,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 → imt_ring-1.6.20}/src/imt_ring.egg-info/PKG-INFO

@@ -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 → imt_ring-1.6.20}/src/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

{imt_ring-1.6.18 → imt_ring-1.6.20}/src/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(