rebase on master

apache · Mar 12, 2024 · 8aac855 · 8aac855
1 parent 767da96
commit 8aac855
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...e/src/test/scala/org/apache/spark/sql/streaming/TransformWithStateInitialStateSuite.scala b/...e/src/test/scala/org/apache/spark/sql/streaming/TransformWithStateInitialStateSuite.scala
@@ -17,7 +17,7 @@
 
 package org.apache.spark.sql.streaming
 
-import org.apache.spark.sql.KeyValueGroupedDataset
+import org.apache.spark.sql.{Encoders, KeyValueGroupedDataset}
 import org.apache.spark.sql.execution.streaming.MemoryStream
 import org.apache.spark.sql.execution.streaming.state.{RocksDBStateStoreProvider, StateStoreMultipleColumnFamiliesNotSupportedException}
 import org.apache.spark.sql.internal.SQLConf
@@ -35,7 +35,7 @@ class StatefulProcessorWithInitialStateTestClass extends StatefulProcessorWithIn
   }
 
   override def init(operatorOutputMode: OutputMode): Unit = {
-    _valState = getHandle.getValueState[Double]("testInit")
+    _valState = getHandle.getValueState[Double]("testInit", Encoders.scalaDouble)
   }
 
   override def close(): Unit = {}