From d8f912b4e114aa3547aed34be68b5f82a396d759 Mon Sep 17 00:00:00 2001 From: Riya Verma Date: Mon, 5 Aug 2024 13:48:02 -0700 Subject: [PATCH] Modify error class name instead of error parameters --- .../org/apache/spark/sql/errors/QueryExecutionErrors.scala | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/errors/QueryExecutionErrors.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/errors/QueryExecutionErrors.scala index d825069e14837..5db9c2863587b 100644 --- a/sql/catalyst/src/main/scala/org/apache/spark/sql/errors/QueryExecutionErrors.scala +++ b/sql/catalyst/src/main/scala/org/apache/spark/sql/errors/QueryExecutionErrors.scala @@ -2629,11 +2629,8 @@ private[sql] object QueryExecutionErrors extends QueryErrorsBase with ExecutionE stateStoreType: String, e: Throwable): Throwable = { new SparkException( - errorClass = "CANNOT_LOAD_STATE_STORE.OUT_OF_MEMORY", - messageParameters = Map( - "stateStoreType" -> stateStoreType, - "stateStoreId" -> stateStoreId - ), + errorClass = s"CANNOT_LOAD_STATE_STORE.${stateStoreType.toUpperCase()}_OUT_OF_MEMORY", + messageParameters = Map("stateStoreId" -> stateStoreId), cause = e ) }