reformat

matsengrp · Nov 15, 2024 · eec4802 · eec4802
1 parent 25173c6
commit eec4802
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/netam/models.py b/netam/models.py
@@ -720,7 +720,9 @@ def forward(self, amino_acid_indices: Tensor, mask: Tensor) -> Tensor:
         if self.output_dim == 1:
             return self.single_value.expand(amino_acid_indices.shape)
         else:
-            return self.single_value.expand(amino_acid_indices.shape + (self.output_dim,))
+            return self.single_value.expand(
+                amino_acid_indices.shape + (self.output_dim,)
+            )
 
 
 class HitClassModel(nn.Module):