fix pagerank API

blmoistawinde · Jan 16, 2023 · 96f951d · 96f951d
1 parent ef197a8
commit 96f951d
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/harvesttext/summary.py b/harvesttext/summary.py
@@ -39,10 +39,8 @@ def get_summary(self, sents, topK=5, stopwords=None, with_importance=False, stan
         for u, v in combinations(range(len(sent_tokens)), 2):
             G.add_edge(u, v, weight=sim_func(sent_tokens[u], sent_tokens[v]))
 
-        try:
-            pr = nx.pagerank_scipy(G)  # sometimes fail to converge
-        except:
-            pr = nx.pagerank_numpy(G)
+        pr = nx.pagerank(G)  # sometimes fail to converge
+
         pr_sorted = sorted(pr.items(), key=lambda x: x[1], reverse=True)
         if not avoid_repeat:
             ret = []