Merge branch 'main' into feature/aherrera/SNOW-2443512

Author: sfc-gh-aherreraaguilar

CHANGELOG.md

@@ -25,6 +25,10 @@
 
 - Improved `DataFrameReader` to return columns in deterministic order when using `INFER_SCHEMA`.
 
+#### Dependency Updates
+
+- Added a dependency on `protobuf<6.34` (was `<6.32`).
+
 ### Snowpark pandas API Updates
 
 #### New Features

recipe/meta.yaml

@@ -62,7 +62,7 @@ requirements:
     - libffi <=3.4.4
     - pyyaml
     # Snowpark IR
-    - protobuf >=3.20,<6.32
+    - protobuf >=3.20,<6.34
     - python-dateutil
     - tzlocal
   run_constrained:

setup.py

@@ -28,7 +28,7 @@
     "pyyaml",
     "cloudpickle>=1.6.0,<=3.1.1,!=2.1.0,!=2.2.0",
     # `protoc` < 3.20 is not able to generate protobuf code compatible with protobuf >= 3.20.
-    "protobuf>=3.20, <6.32",  # Snowpark IR
+    "protobuf>=3.20, <6.34",  # Snowpark IR
     "python-dateutil",  # Snowpark IR
     "tzlocal",  # Snowpark IR
 ]

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")