From 05657dad4161e83b9b3292a477aebf251d5d98af Mon Sep 17 00:00:00 2001
From: Hao Jin <haoj@andrew.cmu.edu>
Date: Fri, 10 Aug 2018 17:21:02 +0000
Subject: [PATCH] switch to safe_load for kubernetes config load

---
 tracker/dmlc_tracker/kubernetes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tracker/dmlc_tracker/kubernetes.py b/tracker/dmlc_tracker/kubernetes.py
index e4b45f87ce..c38eb3399c 100644
--- a/tracker/dmlc_tracker/kubernetes.py
+++ b/tracker/dmlc_tracker/kubernetes.py
@@ -40,7 +40,7 @@ def create_sched_svc_manifest(name, port):
 def create_job_manifest(envs, commands, name, image, template_file):
     if template_file is not None:
         with open( template_file ) as f:
-            job=yaml.load(f)
+            job=yaml.safe_load(f)
             job["metadata"]["name"]=name
             job["spec"]["template"]["metadata"]["labels"]["app"]=name
             job["spec"]["template"]["spec"]["containers"][0]["image"]=image