Skip to content

Commit 108224a

Browse files
committed
fix
1 parent cce4cd1 commit 108224a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

spark/src/main/scala/org/apache/spark/Plugins.scala

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ import org.apache.spark.internal.config.{EXECUTOR_MEMORY, EXECUTOR_MEMORY_OVERHE
2929
import org.apache.spark.sql.internal.StaticSQLConf
3030

3131
import org.apache.comet.{CometConf, CometSparkSessionExtensions}
32+
import org.apache.comet.CometConf.COMET_ENABLE_ONHEAP_MODE
3233

3334
/**
3435
* Comet driver plugin. This class is loaded by Spark's plugin framework. It will be instantiated
@@ -48,7 +49,7 @@ class CometDriverPlugin extends DriverPlugin with Logging with ShimCometDriverPl
4849
logInfo("CometDriverPlugin init")
4950

5051
if (!CometSparkSessionExtensions.isOffHeapEnabled(sc.getConf) &&
51-
!CometConf.COMET_ENABLE_ONHEAP_MODE.get()) {
52+
!sc.getConf.getBoolean(COMET_ENABLE_ONHEAP_MODE.key, false)) {
5253
logWarning(
5354
"Comet plugin is disabled because Spark is not running in off-heap mode. Set " +
5455
s"${CometConf.COMET_ENABLE_ONHEAP_MODE.key}=true to override this restriction " +

0 commit comments

Comments
 (0)