Fix E714 flake8 warning (x8).

This commit is contained in:
Aymeric Augustin
2019-12-21 18:07:03 +01:00
parent 5eab3cf6bc
commit fd2f17a7a1
8 changed files with 8 additions and 8 deletions

View File

@@ -152,7 +152,7 @@ class TFXxxMainLayer(tf.keras.layers.Layer):
# attention_probs has shape bsz x n_heads x N x N
# input head_mask has shape [num_heads] or [num_hidden_layers x num_heads]
# and head_mask is converted to shape [num_hidden_layers x batch x num_heads x seq_length x seq_length]
if not head_mask is None:
if head_mask is not None:
raise NotImplementedError
else:
head_mask = [None] * self.num_hidden_layers