scalastyle

ericm-db · Nov 25, 2024 · 8d860ee · 8d860ee
1 parent 5d5cb65
commit 8d860ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...main/scala/org/apache/spark/sql/execution/streaming/state/RocksDBStateStoreProvider.scala b/...main/scala/org/apache/spark/sql/execution/streaming/state/RocksDBStateStoreProvider.scala
@@ -646,7 +646,7 @@ object RocksDBStateStoreProvider {
       valueSchema: StructType): RocksDBDataEncoder = {
 
     stateStoreEncoding match {
-      case StateStoreEncoding.Avro.toString =>
+      case "avro" =>
         RocksDBStateStoreProvider.dataEncoderCache.get(
           encoderCacheKey,
           new java.util.concurrent.Callable[AvroStateEncoder] {
@@ -656,7 +656,7 @@ object RocksDBStateStoreProvider {
             }
           }
         )
-      case StateStoreEncoding.UnsafeRow.toString =>
+      case "unsaferow" =>
         RocksDBStateStoreProvider.dataEncoderCache.get(
           encoderCacheKey,
           new java.util.concurrent.Callable[UnsafeRowDataEncoder] {