is_picklable: catch AttributeError (addresses #3771) (#4508)

* is_picklable: catch AttributeError (addresses #3771) * edit Co-authored-by: Jirka Borovec <[email protected]> Co-authored-by: chaton <[email protected]>
Lightning-AI · Nov 4, 2020 · 945d6f5 · 945d6f5
1 parent 0d1365c
commit 945d6f5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/utilities/parsing.py b/pytorch_lightning/utilities/parsing.py
@@ -61,7 +61,7 @@ def is_picklable(obj: object) -> bool:
     try:
         pickle.dumps(obj)
         return True
-    except pickle.PicklingError:
+    except (pickle.PicklingError, AttributeError):
         return False