From ed41e4772b161786a481c7f29cf2176168f1a916 Mon Sep 17 00:00:00 2001
From: Daniel King <daniel@mosaicml.com>
Date: Fri, 24 May 2024 17:03:27 -0700
Subject: [PATCH] fix super call

---
 llmfoundry/utils/exceptions.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/llmfoundry/utils/exceptions.py b/llmfoundry/utils/exceptions.py
index 1a434e1cb4..b3f0030311 100644
--- a/llmfoundry/utils/exceptions.py
+++ b/llmfoundry/utils/exceptions.py
@@ -56,8 +56,7 @@ def __init__(self, message: str, **kwargs: Any) -> None:
             setattr(self, key, value)
             self.serializable_attributes.append(key)
 
-    def __str__(self) -> str:
-        return self.error
+        super().__init__(message)
 
     def __reduce__(self):
         if self.__class__ == BaseContextualError: