Fix inconsistent formatting

2e307738 · Gustaf Ahdritz · ef82ecdd · 2e307738
Commit 2e307738 authored Nov 10, 2021 by Gustaf Ahdritz
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 8 deletions

openfold/model/primitives.py openfold/model/primitives.py +2 -8

No files found.
--- a/openfold/model/primitives.py
+++ b/openfold/model/primitives.py
@@ -312,16 +312,10 @@ class GlobalAttention(nn.Module):
        )

        self.linear_k = Linear(
-            c_in,
-            c_hidden,
-            bias=False,
-            init="glorot",
+            c_in, c_hidden, bias=False, init="glorot",
        )
        self.linear_v = Linear(
-            c_in,
-            c_hidden,
-            bias=False,
-            init="glorot",
+            c_in, c_hidden, bias=False, init="glorot",
        )
        self.linear_g = Linear(c_in, c_hidden * no_heads, init="gating")
        self.linear_o = Linear(c_hidden * no_heads, c_in, init="final")