Merge branch 'main' into helmeleegy-fix-replicate-local-env-doctest

Author: sfc-gh-helmeleegy

src/snowflake/snowpark/functions.py

@@ -11749,7 +11749,7 @@ def regr_avgy(y: ColumnOrName, x: ColumnOrName, _emit_ast: bool = True) -> Colum
     Example::
 
         >>> df = session.create_dataframe([[10, 11], [20, 22], [25, None], [30, 35]], schema=["v", "v2"])
-        >>> df = df.group_by("v").agg(regr_avgy(df["v"], df["v2"]).alias("regr_avgy"))
+        >>> df = df.group_by("v").agg(regr_avgy(df["v"], df["v2"]).alias("regr_avgy")).sort("v")
         >>> df.collect()
         [Row(V=10, REGR_AVGY=10.0), Row(V=20, REGR_AVGY=20.0), Row(V=25, REGR_AVGY=None), Row(V=30, REGR_AVGY=30.0)]
     """
@@ -11838,7 +11838,7 @@ def regr_sxx(y: ColumnOrName, x: ColumnOrName, _emit_ast: bool = True) -> Column
     Example::
 
         >>> df = session.create_dataframe([[10, 11], [20, 22], [25, None], [30, 35]], schema=["v", "v2"])
-        >>> df.group_by("v").agg(regr_sxx(col("v"), col("v2")).alias("regr_sxx")).collect()
+        >>> df.group_by("v").agg(regr_sxx(col("v"), col("v2")).alias("regr_sxx")).sort("v").collect()
         [Row(V=10, REGR_SXX=0.0), Row(V=20, REGR_SXX=0.0), Row(V=25, REGR_SXX=None), Row(V=30, REGR_SXX=0.0)]
     """
     y_col = _to_col_if_str(y, "regr_sxx")

src/snowflake/snowpark/session.py

@@ -1794,17 +1794,17 @@ def replicate_local_environment(
 
         Example::
 
-            >>> import sys  # doctest: +SKIP
+            >>> import sys
             >>> from snowflake.snowpark.functions import udf  # doctest: +SKIP
-            >>> import numpy  # doctest: +SKIP
-            >>> import pandas  # doctest: +SKIP
+            >>> import numpy
+            >>> import pandas
             >>> # test_requirements.txt contains "numpy" and "pandas"
-            >>> session.custom_package_usage_config = {"enabled": True, "force_push": True} # Recommended configuration  # doctest: +SKIP
+            >>> session.custom_package_usage_config = {"enabled": True, "force_push": True} # Recommended configuration
             >>> session.replicate_local_environment(ignore_packages={"snowflake-snowpark-python", "snowflake-connector-python", "urllib3", "tzdata", "numpy"}, relax=True)  # doctest: +SKIP
-            >>> @udf  # doctest: +SKIP
-            ... def get_package_name_udf() -> list:  # doctest: +SKIP
-            ...     return [numpy.__name__, pandas.__name__]  # doctest: +SKIP
-            >>> if sys.version_info <= (3, 11):  # doctest: +SKIP
+            >>> @udf
+            ... def get_package_name_udf() -> list:
+            ...     return [numpy.__name__, pandas.__name__]
+            >>> if sys.version_info <= (3, 11):
             ...     session.sql(f"select {get_package_name_udf.name}()").to_df("col1").show()  # doctest: +SKIP
             --------------
             |"COL1"      |