Better progress messages

Knowledge-Graph-Hub · May 20, 2020 · 1a80bce · 1a80bce
1 parent c8596d0
commit 1a80bce
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/kg_covid_19/edges.py b/kg_covid_19/edges.py
@@ -184,6 +184,7 @@ def _generate_negative_edges(nodes_df: pd.DataFrame,
                                        'object': negative_edges['object'],
                                        'relation': relation})
         pbar.update()
+        pbar.set_description("Done making negative edges")
 
     return negative_edges
 
@@ -255,6 +256,7 @@ def make_positive_edges(nodes_df: pd.DataFrame,
         pbar.set_description("Removing test edges from training data")
         train_edges.drop(train_edges.index[test_edges.index], inplace=True)
         pbar.update()
+        pbar.set_description("Done making positive edges")
 
     return [train_edges, test_edges]