From 57426437103d62e4e57d781d2d01588b8c780643 Mon Sep 17 00:00:00 2001
From: Bing Li <bing.li@snowflake.com>
Date: Wed, 7 Feb 2024 16:03:49 -0800
Subject: [PATCH] Support Java 17 when registering SP

---
 .github/workflows/precommit-udf-multiple-jdk.yml                | 2 +-
 .../snowflake/snowpark/internal/UDXRegistrationHandler.scala    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/precommit-udf-multiple-jdk.yml b/.github/workflows/precommit-udf-multiple-jdk.yml
index 3db20fa3..4ac649e1 100644
--- a/.github/workflows/precommit-udf-multiple-jdk.yml
+++ b/.github/workflows/precommit-udf-multiple-jdk.yml
@@ -24,4 +24,4 @@ jobs:
         env:
           PROFILE_PASSWORD: ${{ secrets.PROFILE_PASSWORD }}
       - name: Run test
-        run: mvn -Dgpg.skip test -Dsuites="com.snowflake.snowpark_test.UDTFSuite,com.snowflake.snowpark_test.AlwaysCleanUDFSuite"
+        run: mvn -Dgpg.skip test -Dsuites="com.snowflake.snowpark_test.UDTFSuite,com.snowflake.snowpark_test.AlwaysCleanUDFSuite,com.snowflake.snowpark_test.StoredProcedureSuite"
diff --git a/src/main/scala/com/snowflake/snowpark/internal/UDXRegistrationHandler.scala b/src/main/scala/com/snowflake/snowpark/internal/UDXRegistrationHandler.scala
index 46e34958..18a7ab49 100644
--- a/src/main/scala/com/snowflake/snowpark/internal/UDXRegistrationHandler.scala
+++ b/src/main/scala/com/snowflake/snowpark/internal/UDXRegistrationHandler.scala
@@ -938,7 +938,7 @@ class UDXRegistrationHandler(session: Session) extends Logging {
          |CREATE $tempType PROCEDURE $spName($sqlFunctionArgs)
          |RETURNS $returnSqlType
          |LANGUAGE JAVA
-         |RUNTIME_VERSION = '11'
+         |$getRuntimeVersion
          |$packageSql
          |$importSql
          |target_path='$targetJarStageLocation'