Flake8

fairseq/models/fconv.py

@@ -119,7 +119,7 @@ class FConvEncoder(FairseqEncoder):
             self.projections.append(Linear(in_channels, out_channels)
                                     if in_channels != out_channels else None)
             if kernel_size % 2 == 1:
-               padding = kernel_size //2
+                padding = kernel_size // 2
             else:
                 padding = 0
             self.convolutions.append(

fairseq/models/lstm.py

@@ -102,8 +102,7 @@ class LSTMEncoder(FairseqEncoder):
         self.padding_idx = dictionary.pad()
         self.embed_tokens = Embedding(num_embeddings, embed_dim, self.padding_idx)
         if embed_dict:
-            self.embed_tokens = utils.load_embedding(
-                embed_dict, self.dictionary, self.embed_tokens)
+            self.embed_tokens = utils.load_embedding(embed_dict, self.dictionary, self.embed_tokens)
 
         self.lstm = LSTM(
             input_size=embed_dim,
@@ -195,9 +194,7 @@ class LSTMDecoder(FairseqIncrementalDecoder):
         padding_idx = dictionary.pad()
         self.embed_tokens = Embedding(num_embeddings, embed_dim, padding_idx)
         if embed_dict:
-            self.embed_tokens = utils.load_embedding(
-                embed_dict, self.dictionary, self.embed_tokens)
-
+            self.embed_tokens = utils.load_embedding(embed_dict, self.dictionary, self.embed_tokens)
 
         self.layers = nn.ModuleList([
             LSTMCell(encoder_embed_dim + embed_dim if layer == 0 else embed_dim, embed_dim)

fairseq/utils.py

@@ -254,6 +254,7 @@ def print_embed_overlap(embed_dict, vocab_dict):
     overlap = len(embed_keys & vocab_keys)
     print("| Found {}/{} types in embedding file.".format(overlap, len(vocab_dict)))
 
+
 def parse_embedding(embed_path):
     """Parse embedding text file into a dictionary of word and embedding tensors.
 
@@ -267,12 +268,13 @@ def parse_embedding(embed_path):
     """
     embed_dict = dict()
     with open(embed_path) as f_embed:
-        _ = next(f_embed) #skip header
+        _ = next(f_embed)  # skip header
         for line in f_embed:
             pieces = line.strip().split()
             embed_dict[pieces[0]] = torch.Tensor([float(weight) for weight in pieces[1:]])
     return embed_dict
 
+
 def load_embedding(embed_dict, vocab, embedding):
     for idx in range(len(vocab)):
         token = vocab[idx]
@@ -280,6 +282,7 @@ def load_embedding(embed_dict, vocab, embedding):
             embedding.weight.data[idx] = embed_dict[token]
     return embedding
 
+
 def replace_unk(hypo_str, src_str, alignment, align_dict, unk):
     from fairseq import tokenizer
     # Tokens are strings here