Upgrade to cloudpickle 0.4.1. (#1164)

* Upgrade to cloudpickle 0.4.1. * Catch more general exceptions thrown by cloudpickle.
2026-07-10 06:11:51 +08:00 · 2017-10-28 03:35:35 -05:00
parent 2b6c7af8ad
commit f59867850e
3 changed files with 9 additions and 6 deletions
@@ -299,7 +299,10 @@ class Worker(object):
                                       .format(type(e.example_object)))
                    print(warning_message)
                except (serialization.RayNotDictionarySerializable,
-                        pickle.pickle.PicklingError):
+                        pickle.pickle.PicklingError,
+                        Exception):
+                    # We also handle generic exceptions here because
+                    # cloudpickle can fail with many different types of errors.
                    _register_class(type(e.example_object), use_pickle=True)
                    warning_message = ("WARNING: Falling back to serializing "
                                       "objects of type {} by using pickle. "