black

geoopt · Feb 12, 2019 · 3973338 · 3973338
1 parent 823671a
commit 3973338
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/geoopt/manifolds/stiefel.py b/geoopt/manifolds/stiefel.py
@@ -187,6 +187,6 @@ def _inner(self, x, u, v):
 
     def _retr(self, x, u, t):
         q, r = linalg.batch_linalg.qr(x + u * t)
-        unflip = linalg.batch_linalg.extract_diag(r).sign().add(.5).sign()
+        unflip = linalg.batch_linalg.extract_diag(r).sign().add(0.5).sign()
         q *= unflip[..., None, :]
         return q