Skip to content

Commit f94a622

Browse files
authored
Merge pull request #398 from brs96/arrow-catch-keyboard-interrupt
Arrow graph construct cleanup on KeyboardInterrupt
2 parents cbf4b79 + 722d515 commit f94a622

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

graphdatascience/query_runner/arrow_graph_constructor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def run(self, node_dfs: List[DataFrame], relationship_dfs: List[DataFrame]) -> N
5555
self._send_dfs(relationship_dfs, "relationship")
5656

5757
self._send_action("RELATIONSHIP_LOAD_DONE", {"name": self._graph_name})
58-
except Exception as e:
58+
except (Exception, KeyboardInterrupt) as e:
5959
self._send_action("ABORT", {"name": self._graph_name})
6060

6161
raise e

0 commit comments

Comments
 (0)