Added support for environment specified kernel timeout (kubeflow#1920)

numerology · Aug 22, 2019 · 56160f1 · 56160f1
1 parent 56dfff5
commit 56160f1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/backend/src/apiserver/visualization/server.py b/backend/src/apiserver/visualization/server.py
@@ -16,6 +16,7 @@
 from argparse import Namespace
 import importlib
 import json
+import os
 from pathlib import Path
 from typing import Text
 import shlex
@@ -31,7 +32,7 @@
 parser.add_argument(
     "--timeout",
     type=int,
-    default=100,
+    default=os.getenv('KERNEL_TIMEOUT', 100),
     help="Amount of time in seconds that a visualization can run for before " +
          "being stopped."
 )