Make the is_picklable function more robust (Lightning-AI#17270)

Author: leng-yue

src/lightning/pytorch/CHANGELOG.md

@@ -27,6 +27,8 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 - Disable `torch.inference_mode` with `torch.compile` in PyTorch 2.0 ([#17215](https://github.com/Lightning-AI/lightning/pull/17215))
 
+- Changed the `is_picklable` util function to handle the edge case that throws a `TypeError` ([#17270](https://github.com/Lightning-AI/lightning/pull/17270))
+
 ### Depercated
 
 -

src/lightning/pytorch/utilities/parsing.py

@@ -32,7 +32,7 @@ def is_picklable(obj: object) -> bool:
     try:
         pickle.dumps(obj)
         return True
-    except (pickle.PickleError, AttributeError, RuntimeError):
+    except (pickle.PickleError, AttributeError, RuntimeError, TypeError):
         return False
 
 

tests/tests_pytorch/utilities/test_parsing.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import inspect
+import threading
 
 import pytest
 from torch.jit import ScriptModule
@@ -169,7 +170,7 @@ class UnpicklableClass:
         pass
 
     true_cases = [None, True, 123, "str", (123, "str"), max]
-    false_cases = [unpicklable_function, UnpicklableClass, ScriptModule()]
+    false_cases = [unpicklable_function, UnpicklableClass, ScriptModule(), threading.Lock()]
 
     for case in true_cases:
         assert is_picklable(case) is True