update asJsonNode impl for scala and java

src/main/java/com/snowflake/snowpark_java/types/Variant.java

@@ -377,14 +377,13 @@ public String asJsonString() {
   }
 
   /**
-   * Converts the variant as valid JsonNode. This function allows to read the JSON object directly
-   * rather parsing it as String and then extract the desired value
+   * Return the variant value as a JsonNode. This function allows to read the JSON object directly
+   * as JsonNode from variant column rather parsing it as String
    *
    * <pre>{@code - to get the first value from array for key "a"
    *
    * Variant jv = new Variant("{\"a\": [1, 2], \"b\": \"c\"}");
-   * JsonNode jNode = jv.asJsonNode();
-   * System.out.println(jNode.get("a").get(0));
+   * System.out.println(jv.asJsonNode().get("a").get(0));
    *
    * output
    * 1
@@ -394,7 +393,7 @@ public String asJsonString() {
    * @since 1.14.0
    */
   public JsonNode asJsonNode() {
-    return objectToJsonNode(value);
+    return value;
   }
 
   /**

src/main/scala/com/snowflake/snowpark/types/Variant.scala

@@ -382,20 +382,20 @@ class Variant private[snowpark] (
   }
 
   /**
-   * Converts the variant as valid JsonNode
-   * This function allows to read the JSON object directly rather parsing it as String
-   * and then extract the desired value
+   * Return the variant value as a JsonNode. This function allows to read the JSON object directly
+   * as JsonNode from variant column rather parsing it as String
    * Example - to get the first value from array for key "a"
    * {{{
    *   val sv = new Variant("{\"a\": [1, 2], \"b\": 3, \"c\": \"xyz\"}")
-   *   println(sv.asMap.get("a").get.asJsonNode().get(0))
+   *   println(sv.asJsonNode().get("a").get(0))
    * output
    * 1
    * }}}
+ *
    * @since 1.14.0
    */
   def asJsonNode(): JsonNode = {
-    objectToJsonNode(value)
+    value
   }
 
   /**

src/test/java/com/snowflake/snowpark_test/JavaVariantSuite.java

@@ -370,8 +370,7 @@ public void equalsAndToString() {
   @Test
   public void javaJsonNodeVariantConverter() throws IllegalArgumentException {
     Variant jv = new Variant("{\"a\": [1, 2], \"b\": \"c\"}");
-    JsonNode jNode = jv.asJsonNode();
-    assert (jNode.get("a").isArray());
-    assert (jNode.get("b").asText().equals("c"));
+    assert (jv.asJsonNode().get("a").isArray());
+    assert (jv.asJsonNode().get("b").asText().equals("c"));
   }
 }

src/test/scala/com/snowflake/snowpark_test/ScalaVariantSuite.scala

@@ -314,8 +314,8 @@ class ScalaVariantSuite extends FunSuite {
 
   test("JsonNode") {
     val sv = new Variant("{\"a\": [1, 2], \"b\": 3, \"c\": \"xyz\"}")
-    assert(sv.asMap.get("a").get.asJsonNode().isArray)
-    assert(sv.asMap.get("b").get.asJsonNode().asInt().equals(3))
-    assert(sv.asMap.get("c").get.asJsonNode().asText().equals("xyz"))
+    assert(sv.asJsonNode().get("a").isArray)
+    assert(sv.asJsonNode().get("b").asInt().equals(3))
+    assert(sv.asJsonNode().get("c").asText().equals("xyz"))
   }
 }