Support UDTF with Java 17

snowflakedb · Dec 14, 2023 · a8e6faf · a8e6faf
1 parent 9414376
commit a8e6faf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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.AlwaysCleanUDFSuite"
+        run: mvn -Dgpg.skip test -Dsuites="com.snowflake.snowpark_test.UDTFSuite,com.snowflake.snowpark_test.AlwaysCleanUDFSuite"
diff --git a/src/main/scala/com/snowflake/snowpark/internal/UDXRegistrationHandler.scala b/src/main/scala/com/snowflake/snowpark/internal/UDXRegistrationHandler.scala
@@ -649,7 +649,8 @@ class UDXRegistrationHandler(session: Session) extends Logging {
     } else ""
     val createUdfQuery = s"CREATE $tempType " +
       s"FUNCTION $udfName($sqlFunctionArgs) RETURNS " +
-      s"$returnSqlType LANGUAGE JAVA IMPORTS = ($allImports) HANDLER='$udtfClassName' " +
+      s"$returnSqlType LANGUAGE JAVA $getRuntimeVersion " +
+      s"IMPORTS = ($allImports) HANDLER='$udtfClassName' " +
       s"target_path='$targetJarStageLocation' " + packageSql +
       "AS $$ \n" + code + "\n$$"
     logInfo(s"""