Skip to content

Commit 3e55ddd

Browse files
allow Any values in session_conf, cast to String as done in Thrift backend
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 2caf38d commit 3e55ddd

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/databricks/sql/backend/sea/backend.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import logging
44
import time
55
import re
6-
from typing import Dict, Tuple, List, Optional, Union, TYPE_CHECKING, Set
6+
from typing import Any, Dict, Tuple, List, Optional, Union, TYPE_CHECKING, Set
77

88
from databricks.sql.backend.sea.models.base import ExternalLink, ResultManifest
99
from databricks.sql.backend.sea.utils.constants import (
@@ -49,7 +49,7 @@
4949

5050

5151
def _filter_session_configuration(
52-
session_configuration: Optional[Dict[str, str]]
52+
session_configuration: Optional[Dict[str, Any]]
5353
) -> Optional[Dict[str, str]]:
5454
if not session_configuration:
5555
return None
@@ -59,7 +59,7 @@ def _filter_session_configuration(
5959

6060
for key, value in session_configuration.items():
6161
if key.upper() in ALLOWED_SESSION_CONF_TO_DEFAULT_VALUES_MAP:
62-
filtered_session_configuration[key.lower()] = value
62+
filtered_session_configuration[key.lower()] = str(value)
6363
else:
6464
ignored_configs.add(key)
6565

@@ -183,7 +183,7 @@ def max_download_threads(self) -> int:
183183

184184
def open_session(
185185
self,
186-
session_configuration: Optional[Dict[str, str]],
186+
session_configuration: Optional[Dict[str, Any]],
187187
catalog: Optional[str],
188188
schema: Optional[str],
189189
) -> SessionId:

0 commit comments

Comments
 (0)