small

ipyflow · Feb 24, 2024 · be061c2 · be061c2
1 parent ba80697
commit be061c2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/core/ipyflow/tracing/external_calls/cloudpickle_patch.py b/core/ipyflow/tracing/external_calls/cloudpickle_patch.py
@@ -14,7 +14,8 @@ def _function_reduce(self_, obj) -> None:
 def _patched_function_reduce(
     self_: "CloudPickler", obj: Union[FunctionType, LambdaType]
 ) -> None:
-    return _function_reduce(self_, uninstrument(obj))
+    uninstrumented = uninstrument(obj)
+    return _function_reduce(self_, obj if uninstrumented is None else uninstrumented)
 
 
 def patch_cloudpickle_function_reduce(pickler_cls: Type["CloudPickler"]) -> None: