NO-SNOW: Fix python 3.14 artifact repository test failures (#4170)

Author: sfc-gh-joshi

CHANGELOG.md

@@ -7,6 +7,7 @@
 #### Bug Fixes
 
 - Fixed a bug where CTE optimization incorrectly deduplicated subtrees containing non-deterministic data generation functions (e.g. `uuid_string()`). 
+- Fixed a bug where vectorized UDFs using non-anaconda package repositories did not specify the pandas package by default.
 
 ## 1.49.0 (TBD)
 

src/snowflake/snowpark/_internal/udf_utils.py

@@ -1253,6 +1253,7 @@ def resolve_imports_and_packages(
                     [],
                     artifact_repository=artifact_repository,
                     existing_packages_dict=existing_packages_dict,
+                    include_pandas=is_pandas_udf,
                 )
             )
         elif packages:

src/snowflake/snowpark/functions.py

@@ -11945,7 +11945,7 @@ def regr_sxy(y: ColumnOrName, x: ColumnOrName, _emit_ast: bool = True) -> Column
 
         >>> df = session.create_dataframe([[10, 11], [20, 22], [25, None], [30, 35]], schema=["v", "v2"])
         >>> df = df.filter(df["v2"].is_not_null())
-        >>> df.group_by("v").agg(regr_sxy(df["v"], df["v2"]).alias("regr_sxy")).collect()
+        >>> df.group_by("v").agg(regr_sxy(df["v"], df["v2"]).alias("regr_sxy")).sort("v").collect()
         [Row(V=10, REGR_SXY=0.0), Row(V=20, REGR_SXY=0.0), Row(V=30, REGR_SXY=0.0)]
     """
     y_col = _to_col_if_str(y, "regr_sxy")

tests/integ/test_packaging.py

@@ -22,7 +22,7 @@
 from snowflake.snowpark.functions import call_udf, col, count_distinct, sproc, udf
 from snowflake.snowpark.context import _ANACONDA_SHARED_REPOSITORY
 from snowflake.snowpark.types import DateType, StringType
-from tests.utils import IS_IN_STORED_PROC, TempObjectType, TestFiles, Utils
+from tests.utils import IS_IN_STORED_PROC, TempObjectType, TestFiles, Utils, IS_PY314
 
 pytestmark = pytest.mark.xfail(
     "config.getoption('local_testing_mode', default=False)",
@@ -347,21 +347,23 @@ def test_add_packages_negative(session, caplog):
     assert "InvalidRequirement" in str(ex_info)
 
     session.custom_package_usage_config = {"enabled": True}
-    with patch.object(session, "_is_anaconda_terms_acknowledged", lambda: True):
-        with pytest.raises(RuntimeError, match="Pip failed with return code 1"):
-            session.add_packages("dateutil")
-
-    with patch.object(session, "_is_anaconda_terms_acknowledged", lambda: False):
-        with pytest.raises(RuntimeError, match="Cannot add package dateutil"):
-            session.add_packages("dateutil")
-
-    # Verify multiple errors can be raised at once
-    with patch.object(session, "_is_anaconda_terms_acknowledged", lambda: False):
-        with pytest.raises(
-            RuntimeError,
-            match="Cannot add package dateutil.*Cannot add package functools",
-        ):
-            session.add_packages("dateutil", "functools")
+    # These errors are not raised with Python 3.14 since it uses the pypi repository by default.
+    if not IS_PY314:
+        with patch.object(session, "_is_anaconda_terms_acknowledged", lambda: True):
+            with pytest.raises(RuntimeError, match="Pip failed with return code 1"):
+                session.add_packages("dateutil")
+
+        with patch.object(session, "_is_anaconda_terms_acknowledged", lambda: False):
+            with pytest.raises(RuntimeError, match="Cannot add package dateutil"):
+                session.add_packages("dateutil")
+
+        # Verify multiple errors can be raised at once
+        with patch.object(session, "_is_anaconda_terms_acknowledged", lambda: False):
+            with pytest.raises(
+                RuntimeError,
+                match="Cannot add package dateutil.*Cannot add package functools",
+            ):
+                session.add_packages("dateutil", "functools")
 
     with pytest.raises(ValueError, match="is already added"):
         with caplog.at_level(logging.WARNING):
@@ -464,6 +466,10 @@ def test_add_requirements_twice_should_fail_if_packages_are_different(
     IS_IN_STORED_PROC,
     reason="Subprocess calls are not allowed within stored procedures.",
 )
+@pytest.mark.skipif(
+    IS_PY314,
+    reason="Python 3.14 uses pypi repository",
+)
 def test_add_unsupported_requirements_should_fail_if_custom_packages_upload_enabled_not_switched_on(
     session, resources_path
 ):
@@ -519,6 +525,10 @@ def test_urllib() -> str:
     IS_IN_STORED_PROC,
     reason="Subprocess calls are not allowed within stored procedures.",
 )
+@pytest.mark.skipif(
+    IS_PY314,
+    reason="Python 3.14 uses pypi repository",
+)
 def test_add_unsupported_packages_should_fail_if_custom_packages_upload_enabled_not_switched_on(
     session,
 ):
@@ -991,6 +1001,10 @@ def get_skfuzzy_version() -> str:
     IS_IN_STORED_PROC,
     reason="Subprocess calls are not allowed within stored procedures.",
 )
+@pytest.mark.skipif(
+    IS_PY314,
+    reason="Python 3.14 uses pypi repository",
+)
 def test_add_requirements_unsupported_with_cache_path_negative(
     session, resources_path, temporary_stage
 ):

tests/utils.py

@@ -105,6 +105,8 @@
     "IGNORE_CLIENT_VESRION_IN_STRUCTURED_TYPES_RESPONSE",
 }
 
+IS_PY314 = (sys.version_info.major, sys.version_info.minor) == (3, 14)
+
 
 def current_account(session):
     return session.sql("select CURRENT_ACCOUNT_NAME()").collect()[0][0].upper()