Skip to content

Commit 72100b9

Browse files
let type of rows by List[List[str]] for clarity
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent c21ff5e commit 72100b9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def _convert_json_to_arrow_table(self, rows: List[List]) -> "pyarrow.Table":
130130
names = [col[0] for col in self.description]
131131
return pyarrow.Table.from_arrays(cols, names=names)
132132

133-
def _create_json_table(self, rows: List[List]) -> List[Row]:
133+
def _create_json_table(self, rows: List[List[str]]) -> List[Row]:
134134
"""
135135
Convert raw data rows to Row objects with named columns based on description.
136136
@@ -143,7 +143,7 @@ def _create_json_table(self, rows: List[List]) -> List[Row]:
143143
ResultRow = Row(*[col[0] for col in self.description])
144144
return [ResultRow(*self._convert_json_types(row)) for row in rows]
145145

146-
def fetchmany_json(self, size: int) -> List[List]:
146+
def fetchmany_json(self, size: int) -> List[List[str]]:
147147
"""
148148
Fetch the next set of rows as a columnar table.
149149
@@ -165,7 +165,7 @@ def fetchmany_json(self, size: int) -> List[List]:
165165

166166
return results
167167

168-
def fetchall_json(self) -> List[List]:
168+
def fetchall_json(self) -> List[List[str]]:
169169
"""
170170
Fetch all remaining rows as a columnar table.
171171

0 commit comments

Comments
 (0)