Commit 55dc4842 authored by Myle Ott's avatar Myle Ott
Browse files

Fix tests

parent e89329d6

fairseq/modules/linearized_convolution.py

+1 −0
Changes for fairseq/modules/linearized_convolution.py: 1 added line, 0 removed lines.
Original line number Diff line number Diff line
@@ -5,6 +5,7 @@ 
# the root directory of this source tree. An additional grant of patent rights

# can be found in the PATENTS file in the same directory.



import torch

import torch.nn.functional as F



from fairseq import utils

fairseq/sequence_generator.py

+1 −0
Changes for fairseq/sequence_generator.py: 1 added line, 0 removed lines.
Original line number Diff line number Diff line
@@ -6,6 +6,7 @@ 
# can be found in the PATENTS file in the same directory.



import math



import torch



from fairseq import utils

fairseq/sequence_scorer.py

+2 −0
Changes for fairseq/sequence_scorer.py: 2 added lines, 0 removed lines.
Original line number Diff line number Diff line
@@ -5,6 +5,8 @@ 
# the root directory of this source tree. An additional grant of patent rights

# can be found in the PATENTS file in the same directory.



import torch



from fairseq import utils