Allow colored logging in jupyter notebooks.

facebookresearch · Aug 29, 2020 · 18bfd46 · 18bfd46
1 parent e08125c
commit 18bfd46
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/parlai/utils/logging.py b/parlai/utils/logging.py
@@ -47,6 +47,14 @@
 }
 
 
+def _is_color_friendly():
+    try:
+        __IPYTHON__
+        return True
+    except NameError:
+        return sys.stdout.isatty()
+
+
 # Some functions in this class assume that ':' will be the separator used in
 # the logging formats setup for this class
 class ParlaiLogger(logging.Logger):