diff --git a/yuuno_ipython/ipy_vs/log.py b/yuuno_ipython/ipy_vs/log.py index 92a5c3f..e940c81 100644 --- a/yuuno_ipython/ipy_vs/log.py +++ b/yuuno_ipython/ipy_vs/log.py @@ -58,6 +58,13 @@ def _repr_pretty_(self, p, cycle): class LogWriterFeature(VSFeature): def _push_log_msg(self, level: MessageLevel, msg: str) -> None: + if not isinstance(level, MessageLevel): + try: + level = MessageLevel(level) + except ValueError: + # Handle unknown level, e.g. default or raise + level = MessageLevel.mtInfo + level = level.value if level == MessageLevel.mtDebug: