Skip to content

Commit e84ba7d

Browse files
committed
check types fix
Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>
1 parent 88d8a95 commit e84ba7d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
WaitTimeout,
1515
MetadataCommands,
1616
)
17+
from databricks.sql.thrift_api.TCLIService import ttypes
1718

1819
if TYPE_CHECKING:
1920
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: Union[List[Dict[str, Any]], List["ttypes.TSparkParameter"]],
406407
async_op: bool,
407408
enforce_embedded_schema_correctness: bool,
408409
row_limit: Optional[int] = None,

0 commit comments

Comments
 (0)