diff --git a/src/snowflake/snowpark/_internal/compiler/large_query_breakdown.py b/src/snowflake/snowpark/_internal/compiler/large_query_breakdown.py index b276b9bbaab..3a15002edb8 100644 --- a/src/snowflake/snowpark/_internal/compiler/large_query_breakdown.py +++ b/src/snowflake/snowpark/_internal/compiler/large_query_breakdown.py @@ -3,8 +3,6 @@ # import logging -import os -import tempfile from collections import defaultdict from typing import List, Optional, Tuple @@ -125,7 +123,6 @@ def __init__( self._query_generator = query_generator self.logical_plans = logical_plans self._parent_map = defaultdict(set) - self._tmp_plot_dir = os.path.join(tempfile.gettempdir(), "snowpark_plan_plots") def apply(self) -> List[LogicalPlan]: if is_active_transaction(self.session): diff --git a/src/snowflake/snowpark/session.py b/src/snowflake/snowpark/session.py index f025b2f4610..fefd7109906 100644 --- a/src/snowflake/snowpark/session.py +++ b/src/snowflake/snowpark/session.py @@ -754,7 +754,11 @@ def auto_clean_up_temp_table_enabled(self, value: bool) -> None: @large_query_breakdown_enabled.setter @experimental_parameter(version="1.22.0") def large_query_breakdown_enabled(self, value: bool) -> None: - """Set the value for large_query_breakdown_enabled""" + """Set the value for large_query_breakdown_enabled. When enabled, the client will + automatically detect large query plans and break them down into smaller partitions, + materialize the partitions, and then combine them to execute the query to improve + overall performance. + """ if value in [True, False]: self._conn._telemetry_client.send_large_query_breakdown_telemetry(