shivadarshan-devadiga commented on code in PR #13250:
URL: https://github.com/apache/gluten/pull/13250#discussion_r4261901921


##########
gluten-substrait/src/test/scala/org/apache/spark/sql/WithQueryPlanListener.scala:
##########
@@ -63,10 +67,18 @@ trait WithQueryPlanListener extends SharedSparkSession with 
AnyFunSuiteLike {
   }
 
   override def afterEach(): Unit = {
-    listeners().foreach(listener => 
spark.sessionState.listenerManager.unregister(listener))
-    super.afterEach()
+    // Let any in-flight notifications finish before unregistering, e.g. when 
the test failed.
+    try waitForListenerBus()

Review Comment:
   Thank you for the review @philo-he . Have added braces. Can you PTAL again



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to