File tree Expand file tree Collapse file tree 2 files changed +5
-10
lines changed
src/databricks/sql/backend/sea Expand file tree Collapse file tree 2 files changed +5
-10
lines changed Original file line number Diff line number Diff line change 14
14
WaitTimeout ,
15
15
MetadataCommands ,
16
16
)
17
- from databricks .sql .thrift_api .TCLIService import ttypes
18
17
19
18
if TYPE_CHECKING :
20
19
from databricks .sql .client import Cursor
@@ -406,7 +405,7 @@ def execute_command(
406
405
lz4_compression : bool ,
407
406
cursor : Cursor ,
408
407
use_cloud_fetch : bool ,
409
- parameters : List [ttypes . TSparkParameter ],
408
+ parameters : List [Dict [ str , Any ] ],
410
409
async_op : bool ,
411
410
enforce_embedded_schema_correctness : bool ,
412
411
) -> Union [SeaResultSet , None ]:
@@ -440,9 +439,9 @@ def execute_command(
440
439
for param in parameters :
441
440
sea_parameters .append (
442
441
StatementParameter (
443
- name = param . name ,
444
- value = param . value . stringValue ,
445
- type = param . type ,
442
+ name = param [ " name" ] ,
443
+ value = param [ " value" ] ,
444
+ type = param [ " type" ] if "type" in param else None ,
446
445
)
447
446
)
448
447
Original file line number Diff line number Diff line change @@ -355,11 +355,7 @@ def test_command_execution_advanced(
355
355
"status" : {"state" : "SUCCEEDED" },
356
356
}
357
357
mock_http_client ._make_request .return_value = execute_response
358
- param = Mock ()
359
- param .name = "param1"
360
- param .value = Mock ()
361
- param .value .stringValue = "value1"
362
- param .type = "STRING"
358
+ param = {"name" : "param1" , "value" : "value1" , "type" : "STRING" }
363
359
364
360
with patch .object (sea_client , "get_execution_result" ):
365
361
sea_client .execute_command (
You can’t perform that action at this time.
0 commit comments