Skip to content

GitLab

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

Fix tests

parent e89329d6
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
fairseq/modules/linearized_convolution.py
View file @ 55dc4842
...@@ -5,6 +5,7 @@ ...@@ -5,6 +5,7 @@
# the root directory of this source tree. An additional grant of patent rights # the root directory of this source tree. An additional grant of patent rights
# can be found in the PATENTS file in the same directory. # can be found in the PATENTS file in the same directory.
import torch
import torch.nn.functional as F import torch.nn.functional as F
from fairseq import utils from fairseq import utils
......
fairseq/sequence_generator.py
View file @ 55dc4842
...@@ -6,6 +6,7 @@ ...@@ -6,6 +6,7 @@
# can be found in the PATENTS file in the same directory. # can be found in the PATENTS file in the same directory.
import math import math
import torch import torch
from fairseq import utils from fairseq import utils
......
fairseq/sequence_scorer.py
View file @ 55dc4842
...@@ -5,6 +5,8 @@ ...@@ -5,6 +5,8 @@
# the root directory of this source tree. An additional grant of patent rights # the root directory of this source tree. An additional grant of patent rights
# can be found in the PATENTS file in the same directory. # can be found in the PATENTS file in the same directory.
import torch
from fairseq import utils from fairseq import utils
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment