fixed null softwareProperties issue for clusterProperties

Signed-off-by: cindyyuanjiang <[email protected]>
NVIDIA · Oct 19, 2023 · 3582fb7 · 3582fb7
1 parent 10f289c
commit 3582fb7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/core/src/main/scala/com/nvidia/spark/rapids/tool/profiling/AutoTuner.scala b/core/src/main/scala/com/nvidia/spark/rapids/tool/profiling/AutoTuner.scala
@@ -1035,7 +1035,11 @@ object AutoTuner extends Logging {
     representer.getPropertyUtils.setSkipMissingProperties(true)
     val constructor = new Constructor(classOf[ClusterProperties], new LoaderOptions())
     val yamlObjNested = new Yaml(constructor, representer)
-    Option(yamlObjNested.load(clusterProps).asInstanceOf[ClusterProperties])
+    val clusterPropsOpt = Option(yamlObjNested.load(clusterProps).asInstanceOf[ClusterProperties])
+    if (clusterPropsOpt.isDefined && clusterPropsOpt.get.softwareProperties == null) {
+      clusterPropsOpt.get.softwareProperties = new util.LinkedHashMap[String, String]()
+    }
+    clusterPropsOpt
   }
 
   def loadClusterProps(filePath: String): Option[ClusterProperties] = {