Address review comments: Remove unused code, fix test to write csv

Author: Karuppayya Rajendran

sql/core/src/main/scala/org/apache/spark/sql/execution/SQLExecution.scala

@@ -30,7 +30,7 @@ import org.apache.spark.internal.config.{SPARK_DRIVER_PREFIX, SPARK_EXECUTOR_PRE
 import org.apache.spark.internal.config.Tests.IS_TESTING
 import org.apache.spark.sql.classic.SparkSession
 import org.apache.spark.sql.execution.adaptive.AdaptiveSparkPlanExec
-import org.apache.spark.sql.execution.command.{DataWritingCommandExec, ExecutedCommandExec}
+import org.apache.spark.sql.execution.command.DataWritingCommandExec
 import org.apache.spark.sql.execution.datasources.v2.V2CommandExec
 import org.apache.spark.sql.execution.exchange.ShuffleExchangeLike
 import org.apache.spark.sql.execution.ui.{SparkListenerSQLExecutionEnd, SparkListenerSQLExecutionStart}
@@ -194,9 +194,6 @@ object SQLExecution extends Logging {
                     command.children.flatMap(extractShuffleIds)
                   case dataWritingCommand: DataWritingCommandExec =>
                     extractShuffleIds(dataWritingCommand.child)
-                  case _: ExecutedCommandExec =>
-                    // TODO: Discuss
-                    Seq()
                   case plan =>
                     extractShuffleIds(plan)
                 }

sql/core/src/test/scala/org/apache/spark/sql/execution/QueryExecutionSuite.scala

@@ -351,7 +351,7 @@ class QueryExecutionSuite extends SharedSparkSession {
           val plan = spark.range(100).repartition(10).logicalPlan
           val df = Dataset.ofRows(spark, plan)
           // V1 API write
-          df.write.format("parquet").mode(SaveMode.Overwrite).save(dir.getCanonicalPath)
+          df.write.format("csv").mode(SaveMode.Overwrite).save(dir.getCanonicalPath)
 
           val blockManager = spark.sparkContext.env.blockManager
           assert(blockManager.migratableResolver.getStoredShuffles().isEmpty)