Code style

Author: Mats-SX

graphdatascience/query_runner/arrow_graph_constructor.py

@@ -65,7 +65,7 @@ def run(self, node_dfs: list[DataFrame], relationship_dfs: list[DataFrame]) -> N
             try:
                 self._client.abort(self._graph_name)
             except Exception as abort_exception:
-                if not "No arrow process" in str(abort_exception):
+                if "No arrow process" not in str(abort_exception):
                     self._logger.warning(f"error aborting graph creation: {abort_exception}")
             raise e
 

graphdatascience/query_runner/gds_arrow_client.py

@@ -642,7 +642,7 @@ def send_with_retry() -> dict[str, Any]:
             try:
                 result = client.do_action(
                     action=flight.Action(action_type, json.dumps(meta_data).encode("utf-8")),
-                    options=flight.FlightCallOptions(timeout=10.0)
+                    options=flight.FlightCallOptions(timeout=10.0),
                 )
 
                 # Consume result fully to sanity check and avoid cancelled streams