comments

mosaicml · Dec 15, 2023 · c77161e · c77161e
1 parent 0f46b61
commit c77161e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/llmfoundry/utils/logging_utils.py b/llmfoundry/utils/logging_utils.py
@@ -7,8 +7,15 @@
 class SpecificWarningFilter(logging.Filter):
 
     def __init__(self, message_to_suppress: str):
+        """Filter out a specific warning message based on its content.
+
+        This can be useful for filtering out specific warning messages from third party packages.
+
+        Args:
+            message_to_suppress (str): The warning message to suppress.
+        """
         super().__init__()
         self.message_to_suppress = message_to_suppress
 
-    def filter(self, record: logging.LogRecord):
+    def filter(self, record: logging.LogRecord) -> bool:
         return self.message_to_suppress not in record.getMessage()