black

lezcano · Nov 14, 2021 · 754144e · 754144e
1 parent 8eb7089
commit 754144e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/geotorch/so.py b/geotorch/so.py
@@ -194,7 +194,7 @@ def uniform_init_(tensor):
         # first column by -1 (does not change the measure)
         if n == k:
             mask = (torch.det(q) >= 0.0).float()
-            mask[mask == 0.] = -1.
+            mask[mask == 0.0] = -1.0
             mask = mask.unsqueeze(-1)
             q[..., 0] *= mask
         tensor.copy_(q)