Merge branch 'silence-warnings' into add-GUI

OpenAdaptAI · Jun 7, 2023 · 45a651f · 45a651f
2 parents cc0107f + cf3b7ce
commit 45a651f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/openadapt/config.py b/openadapt/config.py
@@ -19,6 +19,7 @@
     "RECORD_READ_ACTIVE_ELEMENT_STATE": False,
     # TODO: remove?
     "REPLAY_STRIP_ELEMENT_STATE": True,
+    "IGNORE_WARNINGS": True,
 }
 
 

diff --git a/openadapt/utils.py b/openadapt/utils.py
@@ -15,7 +15,7 @@
 import mss.base
 import numpy as np
 
-from openadapt import common
+from openadapt import common, config
 
 
 # TODO: move to config.py
@@ -28,12 +28,19 @@ def configure_logging(logger, log_level):
     log_level_override = os.getenv("LOG_LEVEL")
     log_level = log_level_override or log_level
     logger.remove()
-    logger.add(sys.stderr, level=log_level, filter=good_filter)
+    logger.add(
+        sys.stderr,
+        level=log_level,
+        filter=filter_log_messages if config.IGNORE_WARNINGS else None,
+    )
     logger.debug(f"{log_level=}")
 
 
-def good_filter(data):
-    return not data["message"].endswith("Cannot pickle Objective-C objects')")
+def filter_log_messages(data):
+    messages_to_ignore = [
+        "Cannot pickle Objective-C objects",
+    ]
+    return not any(msg in data["message"] for msg in messages_to_ignore)
 
 
 def row2dict(row, follow=True):