update errors (#357)

PaddlePaddle · Jul 18, 2024 · 4a99909 · 4a99909
1 parent 3002dee
commit 4a99909
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/erniebot/src/erniebot/errors.py b/erniebot/src/erniebot/errors.py
@@ -35,6 +35,12 @@
 class EBError(Exception):
  """Base exception class for the erniebot library."""
 
+ def __init__(self, message: Optional[Any] = None, *args, **kwargs):
+ if message is None:
+ message = str(self.__class__.__name__)
+ message = message
+ super().__init__(message, *args, **kwargs)
+
 
 class ArgumentNotFoundError(EBError):
  """An argument was not found."""