don't hardcode avg

fplll · Jul 10, 2017 · 37cc3c1 · 37cc3c1
1 parent d2fde0e
commit 37cc3c1
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/fpylll/algorithms/bkz_stats.py b/src/fpylll/algorithms/bkz_stats.py
@@ -227,13 +227,9 @@ def __radd__(self, other):
 
     def __sub__(self, other):
         """
-        Return the difference of the averages.
-
+        Return the difference of the two nodes reduced to floats.
         """
-        if not isinstance(other, Statistic):
-            return self.avg - other
-        else:
-            return self.avg - other.avg
+        return float(self) - float(other)
 
     def __float__(self):
         """