From a23d90d4d0123704de83df7954934a49c196b9b4 Mon Sep 17 00:00:00 2001
From: Luis Nino <luis.nino@snowflake.com>
Date: Thu, 3 Oct 2024 09:07:31 -0500
Subject: [PATCH] Fix functions.py

---
 src/snowflake/snowpark/functions.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/snowflake/snowpark/functions.py b/src/snowflake/snowpark/functions.py
index 4fc28e683c..2762b4169b 100644
--- a/src/snowflake/snowpark/functions.py
+++ b/src/snowflake/snowpark/functions.py
@@ -5377,11 +5377,13 @@ def array_remove(array: ColumnOrName, element: ColumnOrLiteral) -> Column:
         -------------
         <BLANKLINE>
 
-        >>> df.select(array_remove(array_remove(array_remove(df.data, 1), 3.1), lit('2').cast(VariantType())).alias("objects")).show()
+        >>> df.select(array_remove(array_remove(df.data, 1), lit('2').cast(VariantType())).alias("objects")).show()
         -------------
         |"OBJECTS"  |
         -------------
-        |[]         |
+        |[          |
+        |  3.1      |
+        |]          |
         -------------
         <BLANKLINE>