Skip to content

Commit

Permalink
Merge pull request #1638 from laws-africa/ranks
Browse files Browse the repository at this point in the history
guard against no ranks when calculating pivot
  • Loading branch information
longhotsummer authored Dec 5, 2023
2 parents 2e5d1d6 + 845978c commit 55c82f6
Showing 1 changed file with 9 additions and 5 deletions.
14 changes: 9 additions & 5 deletions peachjam/graph/ranker.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,11 +53,15 @@ def calculate_ranks(self):
def publish_ranks(self):
"""Store ranks for each work."""
# calculate the pivot as the geometric mean of the ranks
pivot = geometric_mean(x for x in self.ranks if x > 0.0)
log.info(f"Updating pagerank pivot (geometric mean of non-zero ranks): {pivot}")
settings = pj_settings()
settings.pagerank_pivot_value = pivot
settings.save(update_fields=["pagerank_pivot_value"])
ranks = [x for x in self.ranks if x > 0.0]
if ranks:
pivot = geometric_mean(ranks)
log.info(
f"Updating pagerank pivot (geometric mean of non-zero ranks): {pivot}"
)
settings = pj_settings()
settings.pagerank_pivot_value = pivot
settings.save(update_fields=["pagerank_pivot_value"])

updated = []
for work, rank in zip(self.work_ids.keys(), self.ranks):
Expand Down

0 comments on commit 55c82f6

Please sign in to comment.