Arrow graph construct cleanup on keyboardInterrupt

Author: brs96

graphdatascience/query_runner/arrow_graph_constructor.py

@@ -55,7 +55,7 @@ def run(self, node_dfs: List[DataFrame], relationship_dfs: List[DataFrame]) -> N
             self._send_dfs(relationship_dfs, "relationship")
 
             self._send_action("RELATIONSHIP_LOAD_DONE", {"name": self._graph_name})
-        except Exception as e:
+        except (Exception, KeyboardInterrupt) as e:
             self._send_action("ABORT", {"name": self._graph_name})
 
             raise e