Merge pull request #39 from pintaoz-aws/fix_double_quotes

Fix dictionary type cluster parameters in k8s template yaml

Author: rohithn1

launcher/nemo/stages.py

@@ -13,6 +13,7 @@
 # Portions taken from https://github.com/NVIDIA/NeMo-Framework-Launcher, Copyright Nvidia Corporation
 
 
+from ast import literal_eval
 import logging
 import shutil
 from pathlib import Path
@@ -694,13 +695,13 @@ def generate_default_k8s_value_template(self, template_root, cluster_parameters,
         if cluster_parameters.get("namespace", None) is not None:
             values_template.trainingConfig.namespace = cluster_parameters["namespace"]
         if cluster_parameters.get("annotations", None) is not None:
-            values_template.trainingConfig.annotations = cluster_parameters["annotations"]
+            values_template.trainingConfig.annotations = literal_eval(cluster_parameters["annotations"])
         if cluster_parameters.get("priority_class_name", None) is not None:
             values_template.trainingConfig.priorityClassName = cluster_parameters["priority_class_name"]
         if cluster_parameters.get("service_account_name") is not None:
             values_template.trainingConfig.serviceAccountName = cluster_parameters["service_account_name"]
         if cluster_parameters.get("custom_labels", None) is not None:
-            values_template.trainingConfig.customLabels = cluster_parameters["custom_labels"]
+            values_template.trainingConfig.customLabels = literal_eval(cluster_parameters["custom_labels"])
         if cluster_parameters.get("label_selector", None) is not None:
             values_template.trainingConfig.labelSelector = cluster_parameters["label_selector"]
         values_template.trainingConfig.compile = OmegaConf.select(self.cfg, "recipes.run.compile", default=0)