remove excess changes

varun-edachali-dbx · varun-edachali-dbx · commit 01c9b95fb4a9 · 2025-07-15T08:59:21.000+05:30
Signed-off-by: varun-edachali-dbx &lt;varun.edachali@databricks.com&gt;
diff --git a/src/databricks/sql/backend/sea/backend.py b/src/databricks/sql/backend/sea/backend.py
@@ -50,7 +50,7 @@
 
 
 def _filter_session_configuration(
-    session_configuration: Optional[Dict[str, Any]]
+    session_configuration: Optional[Dict[str, str]]
 ) -> Optional[Dict[str, str]]:
     if not session_configuration:
         return None
@@ -60,7 +60,7 @@ def _filter_session_configuration(
 
     for key, value in session_configuration.items():
         if key.upper() in ALLOWED_SESSION_CONF_TO_DEFAULT_VALUES_MAP:
-            filtered_session_configuration[key.lower()] = str(value)
+            filtered_session_configuration[key.lower()] = value
         else:
             ignored_configs.add(key)
 
@@ -188,7 +188,7 @@ def max_download_threads(self) -> int:
 
     def open_session(
         self,
-        session_configuration: Optional[Dict[str, Any]],
+        session_configuration: Optional[Dict[str, str]],
         catalog: Optional[str],
         schema: Optional[str],
     ) -> SessionId:
diff --git a/tests/e2e/common/large_queries_mixin.py b/tests/e2e/common/large_queries_mixin.py
@@ -48,9 +48,7 @@ def fetch_rows(self, cursor, row_count, fetchmany_size):
         "extra_params",
         [
             {},
-            {
-                "use_sea": True,
-            },
+            {"use_sea": True},
         ],
     )
     def test_query_with_large_wide_result_set(self, extra_params):
@@ -83,9 +81,7 @@ def test_query_with_large_wide_result_set(self, extra_params):
         "extra_params",
         [
             {},
-            {
-                "use_sea": True,
-            },
+            {"use_sea": True},
         ],
     )
     def test_query_with_large_narrow_result_set(self, extra_params):
@@ -106,9 +102,7 @@ def test_query_with_large_narrow_result_set(self, extra_params):
         "extra_params",
         [
             {},
-            {
-                "use_sea": True,
-            },
+            {"use_sea": True},
         ],
     )
     def test_long_running_query(self, extra_params):

Original file line number	Diff line number	Diff line change
`@@ -48,9 +48,7 @@ def fetch_rows(self, cursor, row_count, fetchmany_size):`
`48`	`48`	`"extra_params",`
`49`	`49`	`[`
`50`	`50`	`{},`
`51`		`- {`
`52`		`- "use_sea": True,`
`53`		`- },`
	`51`	`+ {"use_sea": True},`
`54`	`52`	`],`
`55`	`53`	`)`
`56`	`54`	`def test_query_with_large_wide_result_set(self, extra_params):`
`@@ -83,9 +81,7 @@ def test_query_with_large_wide_result_set(self, extra_params):`
`83`	`81`	`"extra_params",`
`84`	`82`	`[`
`85`	`83`	`{},`
`86`		`- {`
`87`		`- "use_sea": True,`
`88`		`- },`
	`84`	`+ {"use_sea": True},`
`89`	`85`	`],`
`90`	`86`	`)`
`91`	`87`	`def test_query_with_large_narrow_result_set(self, extra_params):`
`@@ -106,9 +102,7 @@ def test_query_with_large_narrow_result_set(self, extra_params):`
`106`	`102`	`"extra_params",`
`107`	`103`	`[`
`108`	`104`	`{},`
`109`		`- {`
`110`		`- "use_sea": True,`
`111`		`- },`
	`105`	`+ {"use_sea": True},`
`112`	`106`	`],`
`113`	`107`	`)`
`114`	`108`	`def test_long_running_query(self, extra_params):`