snowpark-connect 0.30.0__py3-none-any.whl → 0.30.1__py3-none-any.whl

snowflake/snowpark_connect/config.py

@@ -143,6 +143,9 @@ class GlobalConfig:
         # Control whether repartition(n) on a DataFrame forces splitting into n files during writes
         # This matches spark behavior more closely, but introduces overhead.
         "snowflake.repartition.for.writes": "false",
+        "snowpark.connect.structured_types.fix": str(
+            snowpark.context._enable_fix_2360274
+        ).lower(),
     }
 
     boolean_config_list = [

snowflake/snowpark_connect/proto/snowflake_expression_ext_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

snowflake/snowpark_connect/proto/snowflake_relation_ext_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

snowflake/snowpark_connect/relation/io_utils.py

@@ -4,6 +4,8 @@
 
 from urllib.parse import urlparse
 
+from pyspark.errors.exceptions.base import AnalysisException
+
 CLOUD_PREFIX_TO_CLOUD = {
     "abfss": "azure",
     "wasbs": "azure",
@@ -12,10 +14,28 @@ CLOUD_PREFIX_TO_CLOUD = {
 }
 
 SUPPORTED_COMPRESSION_PER_FORMAT = {
-    "csv": {"AUTO", "GZIP", "BZ2", "BROTLI", "ZSTD", "DEFLATE", "RAW_DEFLATE", "NONE"},
-    "json": {"AUTO", "GZIP", "BZ2", "BROTLI", "ZSTD", "DEFLATE", "RAW_DEFLATE", "NONE"},
-    "parquet": {"AUTO", "LZO", "SNAPPY", "NONE"},
-    "text": {"NONE"},
+    "csv": {
+        "GZIP",
+        "BZ2",
+        "BROTLI",
+        "ZSTD",
+        "DEFLATE",
+        "RAW_DEFLATE",
+        "NONE",
+        "UNCOMPRESSED",
+    },
+    "json": {
+        "GZIP",
+        "BZ2",
+        "BROTLI",
+        "ZSTD",
+        "DEFLATE",
+        "RAW_DEFLATE",
+        "NONE",
+        "UNCOMPRESSED",
+    },
+    "parquet": {"LZO", "SNAPPY", "NONE", "UNCOMPRESSED"},
+    "text": {"NONE", "UNCOMPRESSED"},
 }
 
 
@@ -29,6 +49,43 @@ def is_supported_compression(format: str, compression: str | None) -> bool:
     return compression in supported_compressions_for_format(format)
 
 
+def get_compression_for_source_and_options(
+    source: str, options: dict[str, str], from_read: bool = False
+) -> str | None:
+    """
+    Determines the compression type to use for a given data source and options.
+    Args:
+        source (str): The data source format (e.g., "csv", "json", "parquet", "text").
+        options (dict[str, str]): A dictionary of options that may include a "compression" key.
+    Returns:
+        str: The compression type to use (e.g., "GZIP", "SNAPPY", "NONE").
+    Raises:
+        AnalysisException: If the specified compression is not supported for the given source format.
+    """
+    # From read, we don't have a default compression
+    if from_read and "compression" not in options:
+        return None
+
+    # Get compression from options for proper filename generation
+    default_compression = "NONE" if source != "parquet" else "snappy"
+    compression = options.get("compression", default_compression).upper()
+    if compression == "UNCOMPRESSED":
+        compression = "NONE"
+
+    if not is_supported_compression(source, compression):
+        supported_compressions = supported_compressions_for_format(source)
+        raise AnalysisException(
+            f"Compression {compression} is not supported for {source} format. "
+            + (
+                f"Supported compressions: {sorted(supported_compressions)}"
+                if supported_compressions
+                else "None compression supported for this format."
+            )
+        )
+
+    return compression
+
+
 def get_cloud_from_url(
     url: str,
 ):

snowflake/snowpark_connect/relation/map_row_ops.py

@@ -35,6 +35,9 @@ from snowflake.snowpark_connect.expression.map_expression import (
 )
 from snowflake.snowpark_connect.expression.typer import ExpressionTyper
 from snowflake.snowpark_connect.relation.map_relation import map_relation
+from snowflake.snowpark_connect.utils.identifiers import (
+    split_fully_qualified_spark_name,
+)
 from snowflake.snowpark_connect.utils.telemetry import (
     SnowparkConnectNotImplementedError,
 )
@@ -131,11 +134,19 @@ def map_fillna(
     input_df = input_container.dataframe
 
     if len(rel.fill_na.cols) > 0:
+        if rel.fill_na.cols == ["*"]:
+            # Expand "*" to all columns
+            spark_col_names = input_container.column_map.get_spark_columns()
+        else:
+            spark_col_names = list(rel.fill_na.cols)
+
+        # We don't validate the fully qualified spark name here as fillNa is no-op for structured type colums.
+        # It only works for scalar type columns like float, int, string or bool.
         columns: list[str] = [
             input_container.column_map.get_snowpark_column_name_from_spark_column_name(
-                c
+                split_fully_qualified_spark_name(c)[0]
             )
-            for c in rel.fill_na.cols
+            for c in spark_col_names
         ]
         values = [get_literal_field_and_name(v)[0] for v in rel.fill_na.values]
         if len(values) == 1:

snowflake/snowpark_connect/relation/map_stats.py

@@ -309,9 +309,28 @@ def map_freq_items(rel: relation_proto.Relation) -> DataFrameContainer:
     cols = input_container.column_map.get_snowpark_column_names_from_spark_column_names(
         list(rel.freq_items.cols)
     )
+
+    # handle empty DataFrame case
+    row_count = input_df.count()
+
+    for sp_col_name in cols:
+        spark_col_names.append(
+            f"{input_container.column_map.get_spark_column_name_from_snowpark_column_name(sp_col_name)}_freqItems"
+        )
+
+    if row_count == 0:
+        # If DataFrame is empty, return empty arrays for each column
+        empty_values = [[] for _ in cols]
+        approx_top_k_df = session.createDataFrame([empty_values], spark_col_names)
+        return DataFrameContainer.create_with_column_mapping(
+            dataframe=approx_top_k_df,
+            spark_column_names=spark_col_names,
+            snowpark_column_names=spark_col_names,
+        )
+
     approx_top_k_df = input_df.select(
         *[
-            fn.function("approx_top_k")(fn.col(col), round(input_df.count() / support))
+            fn.function("approx_top_k")(fn.col(col), round(row_count / support))
             for col in cols
         ]
     )
@@ -330,10 +349,6 @@ def map_freq_items(rel: relation_proto.Relation) -> DataFrameContainer:
         for value in approx_top_k_values
     ]
 
-    for sp_col_name in cols:
-        spark_col_names.append(
-            f"{input_container.column_map.get_spark_column_name_from_snowpark_column_name(sp_col_name)}_freqItems"
-        )
     approx_top_k_df = session.createDataFrame([filtered_values], spark_col_names)
 
     return DataFrameContainer.create_with_column_mapping(

snowflake/snowpark_connect/relation/read/map_read.py

@@ -17,6 +17,7 @@ from snowflake.snowpark_connect.config import global_config
 from snowflake.snowpark_connect.dataframe_container import DataFrameContainer
 from snowflake.snowpark_connect.relation.io_utils import (
     convert_file_prefix_path,
+    get_compression_for_source_and_options,
     is_cloud_path,
 )
 from snowflake.snowpark_connect.relation.read.map_read_table import map_read_table
@@ -237,6 +238,14 @@ def _read_file(
     )
     upload_files_if_needed(paths, clean_source_paths, session, read_format)
     paths = [_quote_stage_path(path) for path in paths]
+
+    if read_format in ("csv", "text", "json", "parquet"):
+        compression = get_compression_for_source_and_options(
+            read_format, options, from_read=True
+        )
+        if compression is not None:
+            options["compression"] = compression
+
     match read_format:
         case "csv":
             from snowflake.snowpark_connect.relation.read.map_read_csv import (

snowflake/snowpark_connect/relation/utils.py

@@ -92,6 +92,21 @@ TYPE_MAP_FOR_TO_SCHEMA = {
 }
 
 
+# This mapping is used to map the compression type to the extension of the file.
+FILE_COMPRESSION_TO_EXTENSION = {
+    "GZIP": "gz",
+    "BZ2": "bz2",
+    "BROTLI": "br",
+    "ZSTD": "zst",
+    "DEFLATE": "deflate",
+    "RAW_DEFLATE": "raw_deflate",
+    "SNAPPY": "snappy",
+    "LZO": "lzo",
+    "LZ4": "lz4",
+    "BZIP2": "bz2",
+}
+
+
 def get_df_with_partition_row_number(
     container: DataFrameContainer,
     plan_id: int | None,
@@ -186,13 +201,15 @@ def generate_spark_compatible_filename(
 
     # Add compression if specified and not 'none'
     if compression and compression.lower() not in ("none", "uncompressed"):
-        compression_part = f".{compression.lower()}"
+        compression_part = f".{FILE_COMPRESSION_TO_EXTENSION.get(compression.upper(), compression.lower())}"
     else:
         compression_part = ""
 
     # Add format extension if specified
-    if format_ext:
+    if format_ext == "parquet":
         return f"{base_name}{compression_part}.{format_ext}"
+    elif format_ext is not None and format_ext != "":
+        return f"{base_name}.{format_ext}{compression_part}"
     else:
         return f"{base_name}{compression_part}"
 

snowflake/snowpark_connect/relation/write/map_write.py

@@ -35,9 +35,8 @@ from snowflake.snowpark_connect.config import (
 from snowflake.snowpark_connect.dataframe_container import DataFrameContainer
 from snowflake.snowpark_connect.relation.io_utils import (
     convert_file_prefix_path,
+    get_compression_for_source_and_options,
     is_cloud_path,
-    is_supported_compression,
-    supported_compressions_for_format,
 )
 from snowflake.snowpark_connect.relation.map_relation import map_relation
 from snowflake.snowpark_connect.relation.read.metadata_utils import (
@@ -147,6 +146,11 @@ def map_write(request: proto_base.ExecutePlanRequest):
 
     session: snowpark.Session = get_or_create_snowpark_session()
 
+    # Check for partition hint early to determine precedence over single option
+    partition_hint = (
+        result.partition_hint if hasattr(result, "partition_hint") else None
+    )
+
     # Snowflake saveAsTable doesn't support format
     if (
         write_op.HasField("table")
@@ -176,8 +180,11 @@ def map_write(request: proto_base.ExecutePlanRequest):
             # Generate Spark-compatible filename with proper extension
             extension = write_op.source if write_op.source != "text" else "txt"
 
-            # Get compression from options for proper filename generation
-            compression_option = write_op.options.get("compression", "none")
+            compression = get_compression_for_source_and_options(
+                write_op.source, write_op.options, from_read=False
+            )
+            if compression is not None:
+                write_op.options["compression"] = compression
 
             # Generate Spark-compatible filename or prefix
             # we need a random prefix to support "append" mode
@@ -203,12 +210,12 @@ def map_write(request: proto_base.ExecutePlanRequest):
                 except Exception as e:
                     logger.warning(f"Could not clear directory {write_path}: {e}")
 
-            if should_write_to_single_file:
+            if should_write_to_single_file and partition_hint is None:
                 # Single file: generate complete filename with extension
                 spark_filename = generate_spark_compatible_filename(
                     task_id=0,
                     attempt_number=0,
-                    compression=compression_option,
+                    compression=compression,
                     format_ext=extension,
                 )
                 temp_file_prefix_on_stage = f"{write_path}/{spark_filename}"
@@ -217,29 +224,11 @@ def map_write(request: proto_base.ExecutePlanRequest):
                 spark_filename_prefix = generate_spark_compatible_filename(
                     task_id=0,
                     attempt_number=0,
-                    compression=compression_option,
+                    compression=None,
                     format_ext="",  # No extension for prefix
                 )
                 temp_file_prefix_on_stage = f"{write_path}/{spark_filename_prefix}"
 
-            default_compression = "NONE" if write_op.source != "parquet" else "snappy"
-            compression = write_op.options.get(
-                "compression", default_compression
-            ).upper()
-
-            if not is_supported_compression(write_op.source, compression):
-                supported_compressions = supported_compressions_for_format(
-                    write_op.source
-                )
-                raise AnalysisException(
-                    f"Compression {compression} is not supported for {write_op.source} format. "
-                    + (
-                        f"Supported compressions: {sorted(supported_compressions)}"
-                        if supported_compressions
-                        else "No compression supported for this format."
-                    )
-                )
-
             parameters = {
                 "location": temp_file_prefix_on_stage,
                 "file_format_type": write_op.source
@@ -254,9 +243,6 @@ def map_write(request: proto_base.ExecutePlanRequest):
             # Using the base avoids coupling to exact filenames/prefixes.
             download_stage_path = write_path
 
-            # Check for partition hint early to determine precedence over single option
-            partition_hint = result.partition_hint
-
             # Apply max_file_size for both single and multi-file scenarios
             # This helps control when Snowflake splits files into multiple parts
             if max_file_size:
@@ -314,7 +300,7 @@ def map_write(request: proto_base.ExecutePlanRequest):
                     per_part_prefix = generate_spark_compatible_filename(
                         task_id=part_idx,
                         attempt_number=0,
-                        compression=compression_option,
+                        compression=None,
                         format_ext="",  # prefix only; Snowflake appends extension/counters
                     )
                     part_params["location"] = f"{write_path}/{per_part_prefix}"

snowflake/snowpark_connect/utils/session.py

@@ -120,7 +120,6 @@ def configure_snowpark_session(session: snowpark.Session):
         "PYTHON_SNOWPARK_USE_SCOPED_TEMP_OBJECTS": "false",  # this is required for creating udfs from sproc
         "ENABLE_STRUCTURED_TYPES_IN_SNOWPARK_CONNECT_RESPONSE": "true",
         "QUERY_TAG": f"'{query_tag}'",
-        "FEATURE_INTERVAL_TYPES": "enabled",
     }
 
     session.sql(

snowflake/snowpark_connect/utils/telemetry.py

@@ -11,8 +11,8 @@ from abc import ABC, abstractmethod
 from collections import defaultdict
 from collections.abc import Iterable
 from contextvars import ContextVar
+from dataclasses import dataclass
 from enum import Enum, unique
-from typing import Dict
 
 import google.protobuf.message
 import pyspark.sql.connect.proto.base_pb2 as proto_base
@@ -57,6 +57,7 @@ class TelemetryType(Enum):
 
 class EventType(Enum):
     SERVER_STARTED = "scos_server_started"
+    WARNING = "scos_warning"
 
 
 # global labels
@@ -106,7 +107,16 @@ REDACTED_PLAN_SUFFIXES = [
 ]
 
 
-def _basic_telemetry_data() -> Dict:
+@dataclass
+class TelemetryMessage:
+    """Container for telemetry messages in the processing queue."""
+
+    message: dict
+    timestamp: int
+    is_warning: bool
+
+
+def _basic_telemetry_data() -> dict:
     return {
         **STATIC_TELEMETRY_DATA,
         TelemetryField.KEY_EVENT_ID.value: str(uuid.uuid4()),
@@ -123,9 +133,11 @@ def safe(func):
     def wrap(*args, **kwargs):
         try:
             func(*args, **kwargs)
-        except Exception:
-            # We don't really care if telemetry fails, just want to be safe for the user
-            logger.warning(f"Telemetry operation failed: {func}", exc_info=True)
+        except Exception as e:
+            # report failed operation to telemetry
+            telemetry.send_warning_msg(
+                f"Telemetry operation {func} failed due to exception", e
+            )
 
     return wrap
 
@@ -528,8 +540,8 @@ class Telemetry:
     @safe
     def send_request_summary_telemetry(self):
         if self._not_in_request():
-            logger.warning(
-                "Truing to send request summary telemetry without initializing it"
+            self.send_warning_msg(
+                "Trying to send request summary telemetry without initializing it"
             )
             return
 
@@ -541,14 +553,56 @@ class Telemetry:
         }
         self._send(message)
 
-    def _send(self, msg: Dict) -> None:
+    def send_warning_msg(self, msg: str, e: Exception = None) -> None:
+        # using this within @safe decorator may result in recursive loop
+        try:
+            message = self._build_warning_message(msg, e)
+            if not message:
+                return
+
+            self._send(message, is_warning=True)
+        except Exception:
+            # if there's an exception here, there's nothing we can really do about it
+            pass
+
+    def _build_warning_message(self, warning_msg: str, e: Exception = None) -> dict:
+        try:
+            data = {"warning_message": warning_msg}
+            if e is not None:
+                data["exception"] = repr(e)
+
+            # add session and operation id if available
+            spark_session_id = self._request_summary.get().get("spark_session_id", None)
+            if spark_session_id is not None:
+                data["spark_session_id"] = spark_session_id
+
+            spark_operation_id = self._request_summary.get().get(
+                "spark_operation_id", None
+            )
+            if spark_operation_id is not None:
+                data["spark_operation_id"] = spark_operation_id
+
+            message = {
+                **_basic_telemetry_data(),
+                TelemetryField.KEY_TYPE.value: TelemetryType.TYPE_EVENT.value,
+                TelemetryType.EVENT_TYPE.value: EventType.WARNING.value,
+                TelemetryField.KEY_DATA.value: data,
+            }
+            return message
+        except Exception:
+            return {}
+
+    def _send(self, msg: dict, is_warning: bool = False) -> None:
         """Queue a telemetry message for asynchronous processing."""
         if not self._is_enabled:
             return
 
         timestamp = get_time_millis()
         try:
-            self._message_queue.put_nowait((msg, timestamp))
+            telemetry_msg = TelemetryMessage(
+                message=msg, timestamp=timestamp, is_warning=is_warning
+            )
+            self._message_queue.put_nowait(telemetry_msg)
         except queue.Full:
             # If queue is full, drop the message to avoid blocking
             logger.warning("Telemetry queue is full, dropping message")
@@ -566,13 +620,16 @@ class Telemetry:
         while True:
             try:
                 # block to allow the GIL to switch threads
-                message, timestamp = self._message_queue.get()
-                if timestamp is None and message is None:
-                    # shutdown
+                telemetry_msg = self._message_queue.get()
+                if telemetry_msg is None:
+                    # shutdown signal
                     break
-                self._sink.add_telemetry_data(message, timestamp)
-            except Exception:
-                logger.warning("Failed to add telemetry message to sink", exc_info=True)
+                self._sink.add_telemetry_data(
+                    telemetry_msg.message, telemetry_msg.timestamp
+                )
+            except Exception as e:
+                if not telemetry_msg.is_warning:
+                    self.send_warning_msg("Failed to add telemetry message to sink", e)
             finally:
                 self._message_queue.task_done()
 
@@ -585,7 +642,7 @@ class Telemetry:
             return
 
         try:
-            self._message_queue.put_nowait((None, None))
+            self._message_queue.put_nowait(None)
             # Wait for worker thread to finish
             self._worker_thread.join(timeout=3.0)
         except Exception:
@@ -594,7 +651,7 @@ class Telemetry:
             )
 
 
-def _error_location(e: Exception) -> Dict | None:
+def _error_location(e: Exception) -> dict | None:
     """
     Inspect the exception traceback and extract the file name, line number, and function name
     from the last frame (the one that raised the exception).
@@ -675,7 +732,7 @@ def _protobuf_to_json_with_redaction(
         """Recursively convert protobuf message to dict"""
 
         if not isinstance(msg, google.protobuf.message.Message):
-            logger.warning("Expected a protobuf message, got: %s", type(msg))
+            telemetry.send_warning_msg(f"Expected a protobuf message, got: {type(msg)}")
             return {}
 
         result = {}

snowflake/snowpark_connect/version.py

@@ -2,4 +2,4 @@
 #
 # Copyright (c) 2012-2025 Snowflake Computing Inc. All rights reserved.
 #
-VERSION = (0,30,0)
+VERSION = (0,30,1)

{snowpark_connect-0.30.0.dist-info → snowpark_connect-0.30.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowpark-connect
-Version: 0.30.0
+Version: 0.30.1
 Summary: Snowpark Connect for Spark
 Author: Snowflake, Inc
 License: Apache License, Version 2.0

{snowpark_connect-0.30.0.dist-info → snowpark_connect-0.30.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 snowflake/snowpark_connect/__init__.py,sha256=Sml4x1LTNnxZyw6nnDeJrZWUi3eUAR46Rsw6N-wHUSA,605
 snowflake/snowpark_connect/client.py,sha256=K0PK8aIppghBnY0uKlDOi4WZBUhKxJQ_vRuDpB3YsbQ,2045
 snowflake/snowpark_connect/column_name_handler.py,sha256=eiqUe7YdYFxjyHOAbD9gNCstOIuroZA2INkV4KhqGMg,27407
-snowflake/snowpark_connect/config.py,sha256=jm0jl92VV_CIWJP4cBLcPMJ9Ee14J3oMr-fxF53lbKg,29463
+snowflake/snowpark_connect/config.py,sha256=m2ZPZPaarj2GglLnsuBCgsoJ7SLq3roIC-Jr5YyKl2k,29585
 snowflake/snowpark_connect/constants.py,sha256=FBDxNUxdqWxnf6d5-eanHlYdFFyQqCqvNyZG-uOiO6Q,598
 snowflake/snowpark_connect/control_server.py,sha256=mz3huYo84hgqUB6maZxu3LYyGq7vVL1nv7-7-MjuSYY,1956
 snowflake/snowpark_connect/dataframe_container.py,sha256=0ozyUXrWErzM7Gltnb-i2o5ZyXVVeT_HCqpuYliQXwc,8798
@@ -14,7 +14,7 @@ snowflake/snowpark_connect/start_server.py,sha256=udegO0rk2FeSnXsIcCIYQW3VRlGDjB
 snowflake/snowpark_connect/tcm.py,sha256=ftncZFbVO-uyWMhF1_HYKQykB7KobHEYoyQsYbQj1EM,203
 snowflake/snowpark_connect/type_mapping.py,sha256=hAkOz9_uPi6XnFbd_KcwHqiWoqjt5ht7hlMvD7zR7Cw,45093
 snowflake/snowpark_connect/typed_column.py,sha256=Tavii8b4zMj5IWOvN6tlOVmC80W6eQct0pC_tF2nlhU,3867
-snowflake/snowpark_connect/version.py,sha256=SLJsRwwuumzkgzy7vBGFQsdLIAqh_x7P2gwIM1FD1ao,118
+snowflake/snowpark_connect/version.py,sha256=ihyPxJymHC107aRgWHpDuSBrYCYH4zvut1JtZOdgS5o,118
 snowflake/snowpark_connect/analyze_plan/__init__.py,sha256=xsIE96jDASko3F-MeNf4T4Gg5ufthS8CejeiJDfri0M,76
 snowflake/snowpark_connect/analyze_plan/map_tree_string.py,sha256=RFSNdDwOTXOd3gm_rKRJNyDd65zQpBxBpVOOwmKaFVA,1661
 snowflake/snowpark_connect/error/__init__.py,sha256=oQo6k4zztLmNF1c5IvJLcS99J6RWY9KBTN3RJ2pKimg,249
@@ -392,11 +392,13 @@ snowflake/snowpark_connect/proto/control_pb2.pyi,sha256=DQVcgVtvo1W1tpdi8_ogBO6D
 snowflake/snowpark_connect/proto/control_pb2_grpc.py,sha256=IPHf2f27M1HElF55HH-Kvd8WCFRo6SDj_EjEzFEe_rg,5444
 snowflake/snowpark_connect/proto/snowflake_expression_ext_pb2.py,sha256=Buyg8ACpSbx4A9S-xA5MBAhYhF6oe7YXRs7wCcAU2Wc,3632
 snowflake/snowpark_connect/proto/snowflake_expression_ext_pb2.pyi,sha256=4LLUkliHNthv-H4yFTargdc8FIuytWrg9Ki4JU4Zm1E,4540
+snowflake/snowpark_connect/proto/snowflake_expression_ext_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 snowflake/snowpark_connect/proto/snowflake_rdd_pb2.pyi,sha256=IA_TC6cYM50qTlpSL0UwhmbNkNYgEYuNcZLYTxbGfBc,1780
 snowflake/snowpark_connect/proto/snowflake_relation_ext_pb2.py,sha256=now6RWB80CV5Jko0DBLFMdXYXzvfX30uhHqHb-hKe2o,4886
 snowflake/snowpark_connect/proto/snowflake_relation_ext_pb2.pyi,sha256=4PZkACNf5ulshE8yP2Sl36CXH4UBpIKvZPhzFqeEbsM,7302
+snowflake/snowpark_connect/proto/snowflake_relation_ext_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 snowflake/snowpark_connect/relation/__init__.py,sha256=xsIE96jDASko3F-MeNf4T4Gg5ufthS8CejeiJDfri0M,76
-snowflake/snowpark_connect/relation/io_utils.py,sha256=jBxJZaFde9T6lSWnw23af0Jz_vn0OXD86XVLCJ2og5o,2504
+snowflake/snowpark_connect/relation/io_utils.py,sha256=5FzR34-rEcZkxtLSWpArzpCwr0_Zi9jNY-KM80XBj_k,4266
 snowflake/snowpark_connect/relation/map_aggregate.py,sha256=KElVYO62V3jkU8ldTCfTX2F_atP_GZAozPY1X0W42XQ,19189
 snowflake/snowpark_connect/relation/map_catalog.py,sha256=mcx6An4fqHAxy2OhOC66Xe_0aRtYPDGkBEgMK5CfaXU,5822
 snowflake/snowpark_connect/relation/map_column_ops.py,sha256=cQ9Cde2PMAGXoQWSd984nsp86JCZL7YsAHvfu8Vi50c,49199
@@ -406,22 +408,22 @@ snowflake/snowpark_connect/relation/map_join.py,sha256=I-_jifoCOtahG49maa1Ew4zQ-
 snowflake/snowpark_connect/relation/map_local_relation.py,sha256=VBfwBT75GQUe01UOZptwcYsI7zZxaIMQyTOG6kmVuJ0,15219
 snowflake/snowpark_connect/relation/map_map_partitions.py,sha256=y6WhtacyYRmV8zLv3dQtBP2asijLze363htQbgNyoak,4130
 snowflake/snowpark_connect/relation/map_relation.py,sha256=6RncDhkGXnme6m3sCzOmyfm9x0aM-aqn9H60HY-jdzg,12733
-snowflake/snowpark_connect/relation/map_row_ops.py,sha256=4Zj-fUQlppK1qMc5_kVP3HrcAL_tUQjWFulVTKm-QVQ,34327
+snowflake/snowpark_connect/relation/map_row_ops.py,sha256=76qMXMLisEZpcjLRHkkqufKEIK8NAiGOKcwTI1N-qkY,34877
 snowflake/snowpark_connect/relation/map_sample_by.py,sha256=8ALQbeUsB89sI3uiUFqG3w1A4TtOzOAL4umdKp6-c38,1530
 snowflake/snowpark_connect/relation/map_show_string.py,sha256=-QgOCZ_Kf7qDBN6hd0-eTxXqvVvzoJxw5hGBdk9Upgs,3669
 snowflake/snowpark_connect/relation/map_sql.py,sha256=fDGsl_ZpNo2_rkhXV1cO6i6lnUAnrK-bpiBcKiCnbGU,105507
-snowflake/snowpark_connect/relation/map_stats.py,sha256=AH_yV-o9U2tM925wFjU6LRfwrRA4LhisKjqEfSnnmC8,14002
+snowflake/snowpark_connect/relation/map_stats.py,sha256=xc0ito9lxl3HtYUnWEVuIwV0h3vdD3inSnfmJmztO1g,14493
 snowflake/snowpark_connect/relation/map_subquery_alias.py,sha256=rHgE9XUzuWWkjNPtJz3Sxzz2aFo690paHKZh9frqPXk,1456
 snowflake/snowpark_connect/relation/map_udtf.py,sha256=cfDnbZ3TRJ6eb0EVResu6GL-OwQpaEabWLbrhgWnkRw,13316
 snowflake/snowpark_connect/relation/stage_locator.py,sha256=3PnSmuubtU04wTf4B_kyFK1ag45QXBjzfoKY5mu5Gtk,8925
-snowflake/snowpark_connect/relation/utils.py,sha256=AhE58g0Zz2DWY9gW4JnB_iBU-r4RMnWCj4okQdHSz_4,8398
+snowflake/snowpark_connect/relation/utils.py,sha256=I5June2uch3hsXoQk21y2v_UkLVsBbaJtpV8qEcwIok,8916
 snowflake/snowpark_connect/relation/catalogs/__init__.py,sha256=0yJ5Nfg7SIxudI0P7_U5EWPyiTpkMet8tSq-IwutSZo,265
 snowflake/snowpark_connect/relation/catalogs/abstract_spark_catalog.py,sha256=wujeuKotJZRzpPp_WqVB9TkfprFU2tIXTQJk_eFOR0o,9826
 snowflake/snowpark_connect/relation/catalogs/snowflake_catalog.py,sha256=e9rXL0DOAf4VBZfa-idqehLAEZoCgIoE4IDiLVvR4dQ,28475
 snowflake/snowpark_connect/relation/catalogs/utils.py,sha256=AgiBkK9Xwm9ZyjsKZYK9eTb4YZ9C5dAimD9DLlft-tY,1753
 snowflake/snowpark_connect/relation/read/__init__.py,sha256=5J3IOTKu4Qmenouz1Oz_bUu_4c6KpxtaC63mPUGLyeY,132
 snowflake/snowpark_connect/relation/read/jdbc_read_dbapi.py,sha256=nwvBYwdDTRQc8ljt4CLLaVl5M2uei1TDICJkz7CDSG8,24608
-snowflake/snowpark_connect/relation/read/map_read.py,sha256=y3jCOXzQrL81lz67zYnW1_1ux0uEfYRz0hueYmrv-oc,15460
+snowflake/snowpark_connect/relation/read/map_read.py,sha256=9SBe9h1ZVv9QFQIfFwim2BQODToCIUclNYETZhEjKsA,15770
 snowflake/snowpark_connect/relation/read/map_read_csv.py,sha256=3piGbSdzx5SPr10KCaqF8WrjWDlCeXQlUi3d4cfpeU8,8269
 snowflake/snowpark_connect/relation/read/map_read_jdbc.py,sha256=rfuf2Gp89v-9YFhcXFGdpzmfUgYm1ml0wYN5fdu5HA8,3950
 snowflake/snowpark_connect/relation/read/map_read_json.py,sha256=VWd_g7OeE2dBMxJFvLXt4ntdN9fu8Izerdt3-RGmDYA,13255
@@ -434,7 +436,7 @@ snowflake/snowpark_connect/relation/read/reader_config.py,sha256=PMh1R5IjqqTwiAA
 snowflake/snowpark_connect/relation/read/utils.py,sha256=rIIM6d2WXHh7MLGyHNiRc9tS8b0dmyFQr7rHepIYJOU,4111
 snowflake/snowpark_connect/relation/write/__init__.py,sha256=xsIE96jDASko3F-MeNf4T4Gg5ufthS8CejeiJDfri0M,76
 snowflake/snowpark_connect/relation/write/jdbc_write_dbapi.py,sha256=GI9FyGZuQQNV-6Q8Ob-Xr0im3iAPdH-Jkyx8bjwbOuE,11931
-snowflake/snowpark_connect/relation/write/map_write.py,sha256=Oamzb4KOxt8uJsBxoVj6cAv77NxipVnnAkvKiDabkDA,48265
+snowflake/snowpark_connect/relation/write/map_write.py,sha256=w47pAPxGj1HMZggmxsMsKQEWw47bK3DBct24nMB48aA,47553
 snowflake/snowpark_connect/relation/write/map_write_jdbc.py,sha256=1nOWRgjtZzfRwnSRGFP9V6mqBVlGhSBr2KHGHbe4JMU,1404
 snowflake/snowpark_connect/resources/java_udfs-1.0-SNAPSHOT.jar,sha256=tVyOp6tXxu9nm6SDufwQiGzfH3pnuh_7PowsMZxOolY,9773
 snowflake/snowpark_connect/utils/__init__.py,sha256=xsIE96jDASko3F-MeNf4T4Gg5ufthS8CejeiJDfri0M,76
@@ -451,9 +453,9 @@ snowflake/snowpark_connect/utils/io_utils.py,sha256=j8-t_Vvy0qJhA7EeSYlEqqmc2-os
 snowflake/snowpark_connect/utils/pandas_udtf_utils.py,sha256=3WA_9IVRZL8fnwIHo048LTg62-bPGfCDUZzYd-zjzQQ,7564
 snowflake/snowpark_connect/utils/profiling.py,sha256=ttdHzQUYarvTqJASLNuKFIax7ejO39Tv1mHKl0QjRkg,1519
 snowflake/snowpark_connect/utils/scala_udf_utils.py,sha256=RFDDMmgQ_xBWk98kdfWaw4Hla3ZqYf3UAijU4uAUNdA,23011
-snowflake/snowpark_connect/utils/session.py,sha256=Sd2Oj6CWcR3w1QUt8Kif_o7sxftFOAykjKtFqsKRD_I,7476
+snowflake/snowpark_connect/utils/session.py,sha256=3-76jBcSQFOIJVmXvyClMI2kIz_5PJXWTHE5pcSo_8w,7431
 snowflake/snowpark_connect/utils/snowpark_connect_logging.py,sha256=23bvbALGqixJ3Ap9QWM3OpcKNK-sog2mr9liSmvwqYU,1123
-snowflake/snowpark_connect/utils/telemetry.py,sha256=KcM_fZHlt6v8eoqrfonzgWSZxIQjrZCGkb0RD_PuRd8,23262
+snowflake/snowpark_connect/utils/telemetry.py,sha256=Eh8v7kXbkvfZFcCk_fyMUHM7KjWg7CkeIxYKzvQ1vTo,25267
 snowflake/snowpark_connect/utils/temporary_view_cache.py,sha256=M1czwb-uv0nP9w1meB72NR7cj0iryJc_OYFPztx5XuU,2080
 snowflake/snowpark_connect/utils/udf_cache.py,sha256=AmUFc6iv0UvJF8iH6kAp4BWcZ3O465qyS1pcxf5mQZ8,13342
 snowflake/snowpark_connect/utils/udf_helper.py,sha256=g-TxTs4ARyJWYgADrosfQQG-ykBBQdm1g5opslxJq_E,12563
@@ -467,17 +469,17 @@ snowflake/snowpark_decoder/dp_session.py,sha256=HIr3TfKgYl5zqaGR5xpFU9ZVkcaTB9I8
 snowflake/snowpark_decoder/spark_decoder.py,sha256=EQiCvBiqB736Bc17o3gnYGtcYVcyfGxroO5e1kbe1Co,2885
 snowflake/snowpark_decoder/_internal/proto/generated/DataframeProcessorMsg_pb2.py,sha256=2eSDqeyfMvmIJ6_rF663DrEe1dg_anrP4OpVJNTJHaQ,2598
 snowflake/snowpark_decoder/_internal/proto/generated/DataframeProcessorMsg_pb2.pyi,sha256=aIH23k52bXdw5vO3RtM5UcOjDPaWsJFx1SRUSk3qOK8,6142
-snowpark_connect-0.30.0.data/scripts/snowpark-connect,sha256=yZ94KqbWACxnwV8mpg8NjILvvRNjnF8B3cs3ZFNuIM4,1546
-snowpark_connect-0.30.0.data/scripts/snowpark-session,sha256=NMAHSonTo-nmOZSkQNlszUC0jLJ8QWEDUsUmMe2UAOw,190
-snowpark_connect-0.30.0.data/scripts/snowpark-submit,sha256=Zd98H9W_d0dIqMSkQLdHyW5G3myxF0t4c3vNBt2nD6A,12056
-snowpark_connect-0.30.0.dist-info/licenses/LICENSE-binary,sha256=fmBlX39HwTlBUyiKEznaLZGuxQy-7ndLLG_rTXjF02Y,22916
-snowpark_connect-0.30.0.dist-info/licenses/LICENSE.txt,sha256=Ff9cPv4xu0z7bnMTHzo4vDncOShsy33w4oJMA2xjn6c,11365
-snowpark_connect-0.30.0.dist-info/licenses/NOTICE-binary,sha256=elMF8brgGNJwOz8YdorzBF6-U8ZhR8F-77FfGkZng7U,57843
+snowpark_connect-0.30.1.data/scripts/snowpark-connect,sha256=yZ94KqbWACxnwV8mpg8NjILvvRNjnF8B3cs3ZFNuIM4,1546
+snowpark_connect-0.30.1.data/scripts/snowpark-session,sha256=NMAHSonTo-nmOZSkQNlszUC0jLJ8QWEDUsUmMe2UAOw,190
+snowpark_connect-0.30.1.data/scripts/snowpark-submit,sha256=Zd98H9W_d0dIqMSkQLdHyW5G3myxF0t4c3vNBt2nD6A,12056
+snowpark_connect-0.30.1.dist-info/licenses/LICENSE-binary,sha256=fmBlX39HwTlBUyiKEznaLZGuxQy-7ndLLG_rTXjF02Y,22916
+snowpark_connect-0.30.1.dist-info/licenses/LICENSE.txt,sha256=Ff9cPv4xu0z7bnMTHzo4vDncOShsy33w4oJMA2xjn6c,11365
+snowpark_connect-0.30.1.dist-info/licenses/NOTICE-binary,sha256=elMF8brgGNJwOz8YdorzBF6-U8ZhR8F-77FfGkZng7U,57843
 spark/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spark/connect/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spark/connect/envelope_pb2.py,sha256=7Gc6OUA3vaCuTCIKamb_Iiw7W9jPTcWNEv1im20eWHM,2726
 spark/connect/envelope_pb2.pyi,sha256=VXTJSPpcxzB_dWqVdvPY4KkPhJfh0WmkX7SNHWoLhx0,3358
-snowpark_connect-0.30.0.dist-info/METADATA,sha256=lJFacukQEzIhfl_yoxEgZ34moo89Y65PfoymZFq3DRs,1625
-snowpark_connect-0.30.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-snowpark_connect-0.30.0.dist-info/top_level.txt,sha256=ExnWqVpoTHRG99fu_AxXZVOz8c-De7nNu0yFCGylM8I,16
-snowpark_connect-0.30.0.dist-info/RECORD,,
+snowpark_connect-0.30.1.dist-info/METADATA,sha256=GLLoHHRS1oJ03td7EIC8N3EsPqmw7KH3xaZbNNr2wi4,1625
+snowpark_connect-0.30.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+snowpark_connect-0.30.1.dist-info/top_level.txt,sha256=ExnWqVpoTHRG99fu_AxXZVOz8c-De7nNu0yFCGylM8I,16
+snowpark_connect-0.30.1.dist-info/RECORD,,