Allow bool input for loggers

mosaicml · Jan 23, 2024 · 000a735 · 000a735
1 parent b2a0c03
commit 000a735
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()
     }