Skip to content

Commit 0682761

Browse files
preliminary fix
Signed-off-by: varun-edachali-dbx <[email protected]>
1 parent a74d279 commit 0682761

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

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

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
ResultCompression,
1212
WaitTimeout,
1313
)
14+
from databricks.sql.thrift_api.TCLIService import ttypes
1415

1516
if TYPE_CHECKING:
1617
from databricks.sql.client import Cursor
@@ -402,7 +403,7 @@ def execute_command(
402403
lz4_compression: bool,
403404
cursor: "Cursor",
404405
use_cloud_fetch: bool,
405-
parameters: List[Dict[str, Any]],
406+
parameters: List[ttypes.TSparkParameter],
406407
async_op: bool,
407408
enforce_embedded_schema_correctness: bool,
408409
) -> Union["ResultSet", None]:
@@ -436,9 +437,9 @@ def execute_command(
436437
for param in parameters:
437438
sea_parameters.append(
438439
StatementParameter(
439-
name=param["name"],
440-
value=param["value"],
441-
type=param["type"] if "type" in param else None,
440+
name=param.name,
441+
value=param.value,
442+
type=param.type if param.type else None,
442443
)
443444
)
444445

0 commit comments

Comments
 (0)