Standardize code style

openfold/model/pair_transition.py

@@ -24,6 +24,7 @@ class PairTransition(nn.Module):
     """
     Implements Algorithm 15.
     """
+
     def __init__(self, c_z, n, chunk_size=4):
         """
         Args:
@@ -63,7 +64,7 @@ class PairTransition(nn.Module):
             [*, N_res, N_res, C_z] pair embedding update
         """
         # DISCREPANCY: DeepMind forgets to apply the mask in this module.
-        if(mask is None):
+        if mask is None:
             mask = z.new_ones(z.shape[:-1])
 
         # [*, N_res, N_res, 1]
@@ -73,7 +74,7 @@ class PairTransition(nn.Module):
         z = self.layer_norm(z)
 
         inp = {"z": z, "mask": mask}
-        if(self.chunk_size is not None):
+        if self.chunk_size is not None:
             z = chunk_layer(
                 self._transition,
                 inp,