Support pandas 2.2.2

kfollesdal · kfollesdal · commit 9f59b34ba449 · 2024-07-18T22:30:35.000+02:00
See release note numpy 2.2.2: https://pandas.pydata.org/docs/dev/whatsnew/v2.2.0.html#to-numpy-for-numpy-nullable-and-arrow-types-converts-to-suitable-numpy-dtype
diff --git a/src/databricks/sql/client.py b/src/databricks/sql/client.py
@@ -1164,7 +1164,7 @@ def _convert_arrow_table(self, table):
             timestamp_as_object=True,
         )
 
-        res = df.to_numpy(na_value=None)
+        res = df.to_numpy(na_value=None, dtype='object')
         return [ResultRow(*v) for v in res]
 
     @property

Original file line number	Diff line number	Diff line change
`@@ -1164,7 +1164,7 @@ def _convert_arrow_table(self, table):`
`1164`	`1164`	`timestamp_as_object=True,`
`1165`	`1165`	`)`
`1166`	`1166`
`1167`		`- res = df.to_numpy(na_value=None)`
	`1167`	`+ res = df.to_numpy(na_value=None, dtype='object')`
`1168`	`1168`	`return [ResultRow(*v) for v in res]`
`1169`	`1169`
`1170`	`1170`	`@property`