Merge pull request #36 from ehenneken/papernetwork_bug

round() fix
adsabs · Dec 15, 2021 · e527925 · e527925
2 parents a28cddc + 19e39c5
commit e527925
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vis_services/lib/paper_network.py b/vis_services/lib/paper_network.py
@@ -269,8 +269,8 @@ def get_papernetwork(solr_data, max_groups, weighted=True, equalization=False, d
             scale = sqrt(len(reference_dictionary[papers[i]])*len(reference_dictionary[papers[j]]))
             force = 100*C[ref_papers.get(papers[i]),ref_papers.get(papers[j])] / scale
             if force > 0:
-                link_dict["%s\t%s"%(papers[i],papers[j])] = int(round(force))
-                link_dict["%s\t%s"%(papers[j],papers[i])] = int(round(force))
+                link_dict["%s\t%s"%(papers[i],papers[j])] = math.floor(force+.5)
+                link_dict["%s\t%s"%(papers[j],papers[i])] = math.floor(force+.5)
     # Cut the list of links to the maximum allowed by first sorting by force strength and then cutting by maximum allowed
     if do_cutoff:
         link_dict = _sort_and_cut_results(link_dict)