fix flake8 in the CI

malb · May 8, 2024 · 969828f · 969828f
1 parent a08cb22
commit 969828f
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/estimator/reduction.py b/estimator/reduction.py
@@ -390,7 +390,7 @@ def _short_vectors_sieve(self, beta, d, N=None, B=None, preprocess=True, sieve_d
             self.delta(sieve_dim) ** (sieve_dim - 1) * self.delta(beta) ** (1 - sieve_dim)
         )
 
-        #arbitrary choice
+        # arbitrary choice
         if c > 2**1000:
             c = oo
             return (rho, oo, oo, sieve_dim)
@@ -888,7 +888,7 @@ def short_vectors(self, beta, d, N=None, preprocess=True, B=None, C=5.46, sieve_
         rho = sqrt(4 / 3.0) * RR(
             self.delta(sieve_dim) ** (sieve_dim - 1) * self.delta(beta) ** (1 - sieve_dim)
         )
-        
+
         if N == 1:
             if preprocess:
                 return 1.0, self(beta, d, B=B), 1, beta
@@ -902,10 +902,15 @@ def short_vectors(self, beta, d, N=None, preprocess=True, B=None, C=5.46, sieve_
         c = c0 / floor(c1)
         sieve_cost = C * 2 ** RR((self.NN_AGPS[self.nn]["a"] * sieve_dim + self.NN_AGPS[self.nn]["b"]))
 
-        #arbitrary choice
+        # arbitrary choice
         if c > 2**1000:
-            c = oo
-            return(rho, oo, oo, sieve_dim)
+            # set c = oo
+            return (
+                rho,
+                oo,
+                oo,
+                sieve_dim,
+            )
 
         return (
             rho,