From 5e7c77fb1c18ce173acd9a7432d60e826b2af51b Mon Sep 17 00:00:00 2001
From: Srinivas Lade <srinulade1@gmail.com>
Date: Tue, 1 Apr 2025 11:58:57 -0400
Subject: [PATCH] fix

---
 bodo/decorators.py   | 8 ++++----
 bodo/utils/typing.py | 1 +
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/bodo/decorators.py b/bodo/decorators.py
index 194da17189..98765dae09 100644
--- a/bodo/decorators.py
+++ b/bodo/decorators.py
@@ -598,14 +598,14 @@ def impl(context, builder, sig, args):
             for arg_obj in arg_objs:
                 pyapi.decref(arg_obj)
 
-            # cleanup output
-            if callable(out.cleanup):
-                out.cleanup()
-
             # Error during unboxing
             with builder.if_then(out.is_error):
                 context.call_conv.return_exc(builder)
 
+            # cleanup output
+            if callable(out.cleanup):
+                out.cleanup()
+
             return out.value
 
         self._overload_cache[folded_args] = impl
diff --git a/bodo/utils/typing.py b/bodo/utils/typing.py
index c428d3550f..d00d6a7da9 100644
--- a/bodo/utils/typing.py
+++ b/bodo/utils/typing.py
@@ -2873,6 +2873,7 @@ def _check_objmode_type(val, typ):
             f"Invalid Python output data type specified.\nUser specified:\t{typ}\nValue type:\t{val_typ}"
         )
 
+    print(val)
     return val