additional test fixes

snowflakedb · Dec 12, 2024 · 0f2d9d2 · 0f2d9d2
1 parent 7cc0204
commit 0f2d9d2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/snowflake/snowpark/_internal/analyzer/analyzer.py b/src/snowflake/snowpark/_internal/analyzer/analyzer.py
@@ -847,7 +847,7 @@ def do_resolve_with_resolved_children(
                 logical_plan,
                 logical_plan.left_cols,
                 logical_plan.right_cols,
-                self.session.conf.get("use_constant_subquery_alias", False),
+                bool(self.session.conf.get("use_constant_subquery_alias", False)),
             )
 
         if isinstance(logical_plan, TableFunctionRelation):
@@ -935,7 +935,7 @@ def do_resolve_with_resolved_children(
                 join_condition,
                 match_condition,
                 logical_plan,
-                self.session.conf.get("use_constant_subquery_alias", False),
+                bool(self.session.conf.get("use_constant_subquery_alias", False)),
             )
 
         if isinstance(logical_plan, Sort):

diff --git a/src/snowflake/snowpark/session.py b/src/snowflake/snowpark/session.py
@@ -709,7 +709,7 @@ def __setattr__(self, name, value):
             super().__setattr__(name, value)
 
     @property
-    def large_query_breakdown_complexity_bounds(self):
+    def large_query_breakdown_complexity_bounds(self) -> Tuple[int, int]:
         return (
             self._conf.get("_large_query_breakdown_complexity_lower_bound"),
             self._conf.get("_large_query_breakdown_complexity_upper_bound"),

diff --git a/tests/perf/perf_runner.py b/tests/perf/perf_runner.py
@@ -139,7 +139,7 @@ def compute_plan_height_time(dataframe):
 
     session = Session.builder.configs(CONNECTION_PARAMETERS).create()
     session.sql_simplifier_enabled = args.simplify
-    session._cte_optimization_enabled = args.cte
+    session.cte_optimization_enabled = args.cte
     print("Snowpark Python API Performance Test")
     print("Parameters: ", args)
     try: