From be061c26f6dd27481d2e4041a684ceaa8f6ae542 Mon Sep 17 00:00:00 2001 From: Stephen Macke Date: Sat, 24 Feb 2024 11:32:06 -0800 Subject: [PATCH] small --- core/ipyflow/tracing/external_calls/cloudpickle_patch.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/core/ipyflow/tracing/external_calls/cloudpickle_patch.py b/core/ipyflow/tracing/external_calls/cloudpickle_patch.py index cd3fac4c..e37b469f 100644 --- 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: