feat: support config file for tensorboard [DET-3900]

determined-ai · Aug 27, 2020 · c9d2d6e · c9d2d6e
1 parent 3b10cad
commit c9d2d6e
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/cli/determined_cli/tensorboard.py b/cli/determined_cli/tensorboard.py
@@ -10,7 +10,7 @@
 from determined_common.check import check_eq
 
 from . import render
-from .command import Command, render_event_stream
+from .command import Command, parse_config, render_event_stream
 from .declarative_argparse import Arg, Cmd
 
 Tensorboard = namedtuple(
@@ -37,8 +37,13 @@ def start_tensorboard(args: Namespace) -> None:
         print("Either experiment_ids or trial_ids must be specified.")
         sys.exit(1)
 
-    req_body = {"trial_ids": args.trial_ids, "experiment_ids": args.experiment_ids}
-    resp = api.post(args.master, "tensorboard", body=req_body)
+    config = parse_config(args.config_file, None, [], [])
+    req_body = {
+        "config": config,
+        "trial_ids": args.trial_ids,
+        "experiment_ids": args.experiment_ids,
+    }
+    resp = api.post(args.master, "tensorboard", body=req_body).json()
 
     res_body = resp.json()