Fix yaml load (#140)

* Update admission.py * Update admission.py
Netcracker · Apr 6, 2022 · 77a4388 · 77a4388
1 parent 1e7cad0
commit 77a4388
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kubemarine/admission.py b/kubemarine/admission.py
@@ -370,7 +370,7 @@ def restart_pods_task(cluster, disable_eviction=False):
         first_master.sudo("kubectl uncordon %s" % node["name"], hide=False)
 
     cluster.log.debug("Restarting daemon-sets...")
-    daemon_sets = yaml.load(list(first_master.sudo("kubectl get ds -A -o yaml").values())[0].stdout)
+    daemon_sets = ruamel.yaml.YAML().load(list(first_master.sudo("kubectl get ds -A -o yaml").values())[0].stdout)
     for ds in daemon_sets["items"]:
         first_master.sudo("kubectl rollout restart ds %s -n %s" % (ds["metadata"]["name"], ds["metadata"]["namespace"]))