diff options
Diffstat (limited to 'openbb_platform/core/openbb_core/app/model/abstract/warning.py')
| -rw-r--r-- | openbb_platform/core/openbb_core/app/model/abstract/warning.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/openbb_platform/core/openbb_core/app/model/abstract/warning.py b/openbb_platform/core/openbb_core/app/model/abstract/warning.py index 2e2f778c397..7914b6af7f7 100644 --- a/openbb_platform/core/openbb_core/app/model/abstract/warning.py +++ b/openbb_platform/core/openbb_core/app/model/abstract/warning.py @@ -1,14 +1,19 @@ +"""Module for warnings.""" + from warnings import WarningMessage from pydantic import BaseModel class Warning_(BaseModel): + """Model for Warning.""" + category: str message: str def cast_warning(w: WarningMessage) -> Warning_: + """Cast a warning to a pydantic model.""" return Warning_( category=w.category.__name__, message=str(w.message), @@ -16,4 +21,4 @@ def cast_warning(w: WarningMessage) -> Warning_: class OpenBBWarning(Warning): - pass + """Base class for OpenBB warnings.""" |