From 94474b08c0613be12987bb433fbc3c7d8aac6de7 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Thu, 23 Jan 2025 16:57:01 -0800 Subject: [PATCH] remove logger handler only in notebook --- llama_stack/distribution/library_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/distribution/library_client.py b/llama_stack/distribution/library_client.py index 192667f2ce..b2b290c660 100644 --- a/llama_stack/distribution/library_client.py +++ b/llama_stack/distribution/library_client.py @@ -129,8 +129,8 @@ def initialize(self): import nest_asyncio nest_asyncio.apply() - if not self.skip_logger_removal: - self._remove_root_logger_handlers() + if not self.skip_logger_removal: + self._remove_root_logger_handlers() return asyncio.run(self.async_client.initialize())