From 52360afc4b02d88b5bb23290c66ccfc604e562eb Mon Sep 17 00:00:00 2001 From: Nicholas Garcia Date: Mon, 22 Jan 2024 16:55:52 -0800 Subject: [PATCH] Allow for bool inputs for loggers --- llmfoundry/utils/builders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmfoundry/utils/builders.py b/llmfoundry/utils/builders.py index 75438b895e..3d450eec76 100644 --- a/llmfoundry/utils/builders.py +++ b/llmfoundry/utils/builders.py @@ -220,7 +220,7 @@ def build_callback( def build_logger(name: str, kwargs: Dict[str, Any]) -> LoggerDestination: kwargs_dict = { - k: v if isinstance(v, str) else om.to_container(v, resolve=True) + k: v if isinstance(v, str) or isinstance(v, bool) else om.to_container(v, resolve=True) for k, v in kwargs.items() }