Address comments

apache · Sep 8, 2020 · 6973697 · 6973697
1 parent 15b1673
commit 6973697
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/adaptive/AdaptiveQueryExecSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/adaptive/AdaptiveQueryExecSuite.scala
@@ -891,16 +891,16 @@ class AdaptiveQueryExecSuite
   }
 
   test("SPARK-32753: Only copy tags to node with no tags") {
-    withSQLConf(
-      SQLConf.ADAPTIVE_EXECUTION_ENABLED.key -> "true"
-    ) {
-      spark.range(10).union(spark.range(10)).createOrReplaceTempView("v1")
-
-      val (_, adaptivePlan) = runAdaptiveAndVerifyResult(
-        "SELECT id FROM v1 GROUP BY id DISTRIBUTE BY id")
-      assert(collect(adaptivePlan) {
-        case s: ShuffleExchangeExec => s
-      }.length == 1)
+    withSQLConf(SQLConf.ADAPTIVE_EXECUTION_ENABLED.key -> "true") {
+      withTempView("v1") {
+        spark.range(10).union(spark.range(10)).createOrReplaceTempView("v1")
+
+        val (_, adaptivePlan) = runAdaptiveAndVerifyResult(
+          "SELECT id FROM v1 GROUP BY id DISTRIBUTE BY id")
+        assert(collect(adaptivePlan) {
+          case s: ShuffleExchangeExec => s
+        }.length == 1)
+      }
     }
   }
 }