From 26784147222c5c0167693d3c9673a264abb56c55 Mon Sep 17 00:00:00 2001 From: MSeal Date: Tue, 10 Jun 2014 15:31:40 -0700 Subject: [PATCH] Cleaned up swap statement --- agglomcluster/agglomod.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/agglomcluster/agglomod.py b/agglomcluster/agglomod.py index 8958b78..70f6a82 100644 --- a/agglomcluster/agglomod.py +++ b/agglomcluster/agglomod.py @@ -148,10 +148,8 @@ def run_greedy_clustering(self, quality, reheap_steps=500): def add_pair_to_cost_heap(self, id1, id2): qd = self.quality_difference(id1, id2) - if(id2 < id1): - temp = id1 - id1 = id2 - id2 = temp + if id2 < id1: + id1, id2 = id2, id1 # Negate quality difference (to maximize), AND id1 < id2 heapq.heappush(self.pair_cost_heap, (-qd, id1, id2))