CustomTTYColoredFormatter should inherit TimezoneAware formatter

So that the timestamp format is the same for both.
apache · Dec 28, 2022 · 03e974b · 03e974b
1 parent 69df1c5
commit 03e974b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/airflow/utils/log/colored_log.py b/airflow/utils/log/colored_log.py
@@ -26,6 +26,8 @@
 from colorlog import TTYColoredFormatter
 from colorlog.escape_codes import esc, escape_codes
 
+from airflow.utils.log.timezone_aware import TimezoneAware
+
 DEFAULT_COLORS = {
     "DEBUG": "green",
     "INFO": "",
@@ -38,7 +40,7 @@
 BOLD_OFF = esc("22")
 
 
-class CustomTTYColoredFormatter(TTYColoredFormatter):
+class CustomTTYColoredFormatter(TTYColoredFormatter, TimezoneAware):
     """
     Custom log formatter which extends `colored.TTYColoredFormatter`
     by adding attributes to message arguments and coloring error