Force refresh of progress bar

Previously users reported it got stuck at 0% in jupyter notebooks

Author: FlorentinD

graphdatascience/query_runner/arrow_graph_constructor.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import concurrent
 import json
 import math
@@ -108,6 +110,8 @@ def _send_df(self, df: DataFrame, entity_type: str, pbar: tqdm[NoReturn]) -> Non
             for partition in batches:
                 writer.write_batch(partition)
                 pbar.update(partition.num_rows)
+        # Force a refresh to avoid the progress bar getting stuck at 0%
+        pbar.refresh()
 
     def _send_dfs(self, dfs: List[DataFrame], entity_type: str) -> None:
         desc = "Uploading Nodes" if entity_type == "node" else "Uploading Relationships"

graphdatascience/query_runner/neo4j_query_runner.py

@@ -261,6 +261,7 @@ def _log(self, job_id: str, future: "Future[Any]", database: Optional[str] = Non
 
         for pbar in pbars.values():
             pbar.update(100 - pbar.n)
+            pbar.refresh()
 
     def set_database(self, database: str) -> None:
         self._database = database