reshuffle pickling

OstapHEP · Apr 24, 2024 · eb24388 · eb24388
1 parent ba650ed
commit eb24388
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/ostap/io/pickling.py b/ostap/io/pickling.py
@@ -63,7 +63,7 @@ def pickles ( obj ) :
     try:
         pkl = loads ( dumps ( obj ) )
         return pkl == obj 
-    except ( PicklingError, UnpicklingError ) : 
+    except ( PicklingError, UnpicklingError , AttributeError ) : 
         return False
 
 # =============================================================================
@@ -75,7 +75,7 @@ def check ( obj ):
     fail = True
     try:
         _obj = dumps(obj,)
-    except PicklingError :
+    except ( PicklingError , AttributeError ) :
         return None 
     ## 
     msg = "%s -c import pickle; print(pickle.loads(%s))" %  ( python , repr ( _obj ) )

diff --git a/ostap/parallel/parallel.py b/ostap/parallel/parallel.py
@@ -128,8 +128,11 @@
     worker = 'GAUDI'
 
 # =============================================================================
-## check if object can be pickled 
+## check if object can be pickled & nupickled 
 def pickles ( obj ) :
+    """ Check if object can be pickled & unpiickled
+    """
+    print ( 'PICKLES:' , dill, WorkManager, worker ) 
     if dill and WorkManager and worker == 'PATHOS' :
         return dill.pickles ( obj )
     from ostap.io.pickling import pickles as _pickles
@@ -140,6 +143,7 @@ def pickles ( obj ) :
 def check ( obj ):
     """Check pickling of an object across another process
     """
+    print ( 'CHECK:' , dill, WorkManager, worker ) 
     if dill and WorkManager and worker == 'PATHOS' :
         return dill.check ( obj )
     from ostap.io.pickling import check  as _check