We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ddc2940 commit ba72da0Copy full SHA for ba72da0
1 file changed
src/databricks/sql/client.py
@@ -1408,8 +1408,12 @@ def _convert_arrow_table(self, table: "pyarrow.Table"):
1408
ResultRow = Row(*column_names)
1409
1410
if self.connection.disable_pandas is True:
1411
+ start_time = time.time()
1412
columns_as_lists = [col.to_pylist() for col in table.itercolumns()]
- return [ResultRow(*row) for row in zip(*columns_as_lists)]
1413
+ res = [ResultRow(*row) for row in zip(*columns_as_lists)]
1414
+ end_time = time.time()
1415
+ print(f"Time taken to convert arrow table to list: {end_time - start_time} seconds")
1416
+ return res
1417
1418
# Need to use nullable types, as otherwise type can change when there are missing values.
1419
# See https://arrow.apache.org/docs/python/pandas.html#nullable-types
0 commit comments