Small bug fix

Author: teschlg

kryptools/lat.py

@@ -136,7 +136,8 @@ def compute_weights(self, start: int = 0) -> None:
     def weight_reduce(self) -> bool:
         "Pairwise reducte all weights of the basis vectors."
         reduced = False
-        for j in range(1, self.len):
+        j = 1
+        while j < self.len:
             for i in range(j - 1, -1, -1):
                 r = round(self.Mu[j][i])
                 if not r:
@@ -150,13 +151,15 @@ def weight_reduce(self) -> bool:
                 Nuj = self.norm2(self.U[j])
                 if not Nuj:
                     self.delete_vector(j)
-                    continue
+                    break
                 if self.fraction:
                     Nuj = Fraction(Nuj)
                 for k in range(self.len):
                     if j != k:
                         self.Mu[k][j] = (self.Nu[j] * self.Mu[k][j] - r * self.Nu[i] * self.Mu[k][i]) / Nuj
                 self.Nu[j] = Nuj
+            else:
+                j += 1
         return reduced
 
     def hermite(self) -> None:

kryptools/lat2.py

@@ -4,8 +4,7 @@
 
 from math import prod, floor, inf
 from itertools import product
-from random import choice, sample
-from .la import Matrix, eye, zeros
+from .la import Matrix, eye
 from .lat import Lattice
 
 
@@ -78,7 +77,7 @@ def babai_plane_bnd(U: Matrix, p=2) -> float:
 def lll(V: Matrix, delta: float = 0.75, sort: bool = True) -> Matrix:
     "LLL algorithm for lattice reduction."
     lat = Lattice(V)
-    lat.lll()
+    lat.lll(delta = delta)
     if sort:
         lat.sort()
     return lat.basis()