cli: change torchx color to blue and fix for closed stdout

Author: d4l3k

torchx/cli/colors.py

@@ -8,13 +8,15 @@
 import sys
 
 # only print colors if outputting directly to a terminal
-if sys.stdout.isatty():
+if not sys.stdout.closed and sys.stdout.isatty():
     GREEN = "\033[32m"
+    BLUE = "\033[34m"
     ORANGE = "\033[38:2:238:76:44m"
     GRAY = "\033[2m"
     ENDC = "\033[0m"
 else:
     GREEN = ""
     ORANGE = ""
+    BLUE = ""
     GRAY = ""
     ENDC = ""

torchx/cli/main.py

@@ -17,7 +17,7 @@
 from torchx.cli.cmd_run import CmdBuiltins, CmdRun
 from torchx.cli.cmd_runopts import CmdRunopts
 from torchx.cli.cmd_status import CmdStatus
-from torchx.cli.colors import ENDC, GRAY, ORANGE
+from torchx.cli.colors import ENDC, GRAY, BLUE
 from torchx.util.entrypoints import load_group
 
 
@@ -97,7 +97,7 @@ def run_main(subcmds: Dict[str, SubCommand], argv: List[str] = sys.argv[1:]) ->
     args = parser.parse_args(argv)
     logging.basicConfig(
         level=args.log_level,
-        format=f"{ORANGE}torchx{ENDC} {GRAY}%(asctime)s %(levelname)-8s{ENDC} %(message)s",
+        format=f"{BLUE}torchx{ENDC} {GRAY}%(asctime)s %(levelname)-8s{ENDC} %(message)s",
         datefmt="%Y-%m-%d %H:%M:%S",
     )
     if "func" not in args: