Skip to content

Commit 0b3e91d

Browse files
stronger typing of json utility func s
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent ec500b6 commit 0b3e91d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/databricks/sql/result_set.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -499,7 +499,7 @@ def __init__(
499499
arrow_schema_bytes=execute_response.arrow_schema_bytes or b"",
500500
)
501501

502-
def _convert_json_to_arrow(self, rows):
502+
def _convert_json_to_arrow(self, rows: List) -> "pyarrow.Table":
503503
"""
504504
Convert raw data rows to Arrow table.
505505
"""
@@ -544,7 +544,7 @@ def _convert_json_types(self, rows: List) -> List:
544544

545545
return converted_rows
546546

547-
def _create_json_table(self, rows):
547+
def _create_json_table(self, rows: List) -> List[Row]:
548548
"""
549549
Convert raw data rows to Row objects with named columns based on description.
550550
Also converts string values to appropriate Python types based on column metadata.

0 commit comments

Comments
 (0)