Set environment variable to allow TB profiler plugin to check if it i…

…s running in VS Code (#15711)
microsoft · Mar 23, 2021 · 58df320 · 58df320
1 parent a71e546
commit 58df320
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pythonFiles/tensorboard_launcher.py b/pythonFiles/tensorboard_launcher.py
@@ -1,10 +1,12 @@
 import time
 import sys
+import os
 from tensorboard import default
 from tensorboard import program
 
 
 def main(logdir):
+    os.environ["VSCODE_TENSORBOARD_LAUNCH"] = "1"
     tb = program.TensorBoard(
         default.get_plugins(),
         program.get_default_assets_zip_provider(),