imt-ring 1.6.36__py3-none-any.whl → 1.6.37__py3-none-any.whl

{imt_ring-1.6.36.dist-info → imt_ring-1.6.37.dist-info}/METADATA

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

@@ -83,10 +83,10 @@ ring/utils/hdf5.py,sha256=BzXwVypZmEZeHVgeGZ78YYdi10NEQtnPhdrb8dQAXo0,5856
 ring/utils/normalizer.py,sha256=67L2BU1MRsMT4pD41ta3JJMppLN0ozFmnwrmXDtnqrQ,1698
 ring/utils/path.py,sha256=zRPfxYNesvgefkddd26oar6f9433LkMGkhp9dF3rPUs,1926
 ring/utils/randomize_sys.py,sha256=G_vBIo0OwQkXL2u0djwbaoaeb02C4LQCTNNloOYIU2M,3699
-ring/utils/utils.py,sha256=tJaWXLGOTwkxJQj2l23dX97wO3aZYhM2qd7eNuMRs84,6907
+ring/utils/utils.py,sha256=gKwOXLxWraeZfX6EbBcg3hkq30DcXN0mcRUeOSTNiMo,7336
 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.36.dist-info/METADATA,sha256=a-uW_s0jWJEBX9kW1q36Br4SNXPK7eGIVhlsyKDWruE,4251
-imt_ring-1.6.36.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-imt_ring-1.6.36.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
-imt_ring-1.6.36.dist-info/RECORD,,
+imt_ring-1.6.37.dist-info/METADATA,sha256=NN6c8jI6u0PjCtw3ZXs9Ktk6ODPI9igC-qyS1aUIUsI,4251
+imt_ring-1.6.37.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+imt_ring-1.6.37.dist-info/top_level.txt,sha256=EiT790-lAyi8iwTzJArH3f2k77rwhDn00q-4PlmvDQo,5
+imt_ring-1.6.37.dist-info/RECORD,,

{imt_ring-1.6.36.dist-info → imt_ring-1.6.37.dist-info}/WHEEL

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

ring/utils/utils.py

@@ -1,5 +1,6 @@
 from importlib import import_module as _import_module
 import io
+from pathlib import Path
 import pickle
 import random
 from typing import Optional
@@ -152,13 +153,28 @@ def import_lib(
 
 def pickle_save(obj, path, overwrite: bool = False):
     path = parse_path(path, extension="pickle", file_exists_ok=overwrite)
-    with open(path, "wb") as file:
-        pickle.dump(obj, file, protocol=5)
+    try:
+        with open(path, "wb") as file:
+            pickle.dump(obj, file, protocol=5)
+    except OSError as e:
+        print(
+            f"saving with `pickle` throws exception {e}. "
+            + "Attempting to save using `joblib`"
+        )
+        path = parse_path(path, extension="joblib", file_exists_ok=overwrite)
+        import joblib
+
+        joblib.dump(obj, path)
 
 
 def pickle_load(
     path,
 ):
+    if Path(path).suffix == ".joblib":
+        import joblib
+
+        return joblib.load(path)
+
     path = parse_path(path, extension="pickle", require_is_file=True)
     with open(path, "rb") as file:
         obj = pickle.load(file)