From c098f0c314c6153ab3532266d45faee55b343320 Mon Sep 17 00:00:00 2001 From: xmoforf Date: Tue, 27 May 2025 11:31:40 -0400 Subject: [PATCH] if message level is passed as int, convert it to MessageLevel --- yuuno_ipython/ipy_vs/log.py | 7 +++++++ 1 file changed, 7 insertions(+) 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: