From cdfd70554592f0d698f4d24fc9c41b2b35ee3339 Mon Sep 17 00:00:00 2001
From: Bing Li <63471091+sfc-gh-bli@users.noreply.github.com>
Date: Thu, 8 Feb 2024 09:59:40 -0800
Subject: [PATCH] SNOW-1012646 Support Java 17 when Registering SP (#83)

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'