address feedback

snowflakedb · Aug 28, 2024 · 7073a83 · 7073a83
1 parent 8a5f2b6
commit 7073a83
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -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(