diff --git a/streamingpro-mlsql/src/main/java/tech/mlsql/ets/Ray.scala b/streamingpro-mlsql/src/main/java/tech/mlsql/ets/Ray.scala index db94422d5..41e38f634 100644 --- a/streamingpro-mlsql/src/main/java/tech/mlsql/ets/Ray.scala +++ b/streamingpro-mlsql/src/main/java/tech/mlsql/ets/Ray.scala @@ -128,7 +128,7 @@ class Ray(override val uid: String) extends SQLAlg with VersionCompatibility wit tempdf.rdd.mapPartitions { iter => - val host: String = if (SparkEnv.get == null || MLSQLSparkUtils.rpcEnv().address == null) { + val host: String = if (SparkEnv.get == null || MLSQLSparkUtils.blockManager == null || MLSQLSparkUtils.blockManager.blockManagerId == null) { WriteLog.write(List("Ray: Cannot get MLSQLSparkUtils.rpcEnv().address, using NetTool.localHostName()").iterator, Map("PY_EXECUTE_USER" -> owner)) NetTool.localHostName()