From 8d860eeaf218753d31ad14f4682033cbfa82701f Mon Sep 17 00:00:00 2001
From: Eric Marnadi <eric.marnadi@databricks.com>
Date: Mon, 25 Nov 2024 13:06:17 -0800
Subject: [PATCH] scalastyle

---
 .../execution/streaming/state/RocksDBStateStoreProvider.scala | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/state/RocksDBStateStoreProvider.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/state/RocksDBStateStoreProvider.scala
index 8183a9a1cf7a8..0e82f34d0e9a2 100644
--- a/sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/state/RocksDBStateStoreProvider.scala
+++ b/sql/core/src/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] {