Sanitize table only for numpy backend

Author: FlorentinD

changelog.md

@@ -21,6 +21,6 @@
 * Improve progress bar by showing the description of the currently running task.
 * Allow passing the optional graph filter also as type `str` to `gds.graph.list()` instead of only `Graph`.
 * Listen and to SIGINT and SIGTERM and interrupt projection and write-backs based on GDS Session. Note this only works if the query runs in the main thread.
-
+* Use arrow-based DataFrame backend for arrow-based operations and pandas>2.0
 
 ## Other changes

graphdatascience/query_runner/gds_arrow_client.py

@@ -676,11 +676,10 @@ def _do_get(
             message=r"Passing a BlockManager to DataFrame is deprecated",
         )
 
-        arrow_table = self._sanitize_arrow_table(arrow_table)
-
         if ServerVersion.from_string(pandas.__version__) >= ServerVersion(2, 0, 0):
             return arrow_table.to_pandas(types_mapper=pandas.ArrowDtype)  # type: ignore
         else:
+            arrow_table = self._sanitize_arrow_table(arrow_table)
             return arrow_table.to_pandas()  # type: ignore
 
     def __enter__(self) -> GdsArrowClient: