check for message

ploomber · Dec 10, 2023 · c047cc7 · c047cc7
1 parent 6f20b6b
commit c047cc7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/ploomber_core/exceptions.py b/src/ploomber_core/exceptions.py
@@ -87,6 +87,8 @@ def _add_community_link(e):
     elif COMMUNITY not in e.args[0]:
         message = e.args[0] + COMMUNITY
         e.args = (message,)
+        if hasattr(e, "message"):
+            e.message = message
 
     return e