diff --git a/tools/src/main/scala/org/apache/spark/sql/rapids/tool/AppBase.scala b/tools/src/main/scala/org/apache/spark/sql/rapids/tool/AppBase.scala index 74663f066f8..a978b05680a 100644 --- a/tools/src/main/scala/org/apache/spark/sql/rapids/tool/AppBase.scala +++ b/tools/src/main/scala/org/apache/spark/sql/rapids/tool/AppBase.scala @@ -137,7 +137,7 @@ abstract class AppBase( def getPlanMetaWithSchema(planInfo: SparkPlanInfo): Seq[SparkPlanInfo] = { val childRes = planInfo.children.flatMap(getPlanMetaWithSchema(_)) - if (planInfo.metadata.contains("ReadSchema")) { + if (planInfo.metadata != null && planInfo.metadata.contains("ReadSchema")) { childRes :+ planInfo } else { childRes