From c047cc7a41db5273943129659434c6b3da15a6ca Mon Sep 17 00:00:00 2001
From: neelasha23 <neelasha.sen@gmail.com>
Date: Sat, 9 Dec 2023 21:37:34 +0530
Subject: [PATCH] check for message

---
 src/ploomber_core/exceptions.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/ploomber_core/exceptions.py b/src/ploomber_core/exceptions.py
index 301c304..9ac3b46 100644
--- 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