rebase to latest master

apache · Mar 5, 2024 · fa10594 · fa10594
1 parent cbed9c6
commit fa10594
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/streaming/TransformWithMapStateSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/streaming/TransformWithMapStateSuite.scala
@@ -28,12 +28,10 @@ case class InputMapRow(key: String, action: String, value: (String, String))
 class TestMapStateProcessor
   extends StatefulProcessor[String, InputMapRow, (String, String, String)] {
 
-  @transient var _processorHandle: StatefulProcessorHandle = _
   @transient var _mapState: MapState[String, String] = _
 
-  override def init(handle: StatefulProcessorHandle, outputMode: OutputMode): Unit = {
-    _processorHandle = handle
-    _mapState = handle.getMapState("sessionState", Encoders.STRING)
+  override def init(outputMode: OutputMode): Unit = {
+    _mapState = getHandle.getMapState("sessionState", Encoders.STRING)
   }
 
   override def handleInputRows(