Commit 29c82741 authored by Myle Ott's avatar Myle Ott Committed by Sergey Edunov
Browse files

Fix tests and flake8

parent b9f2d427

preprocess.py

+1 −0
Changes for preprocess.py: 1 added line, 0 removed lines.
Original line number Diff line number Diff line
@@ -39,6 +39,7 @@ def get_parser(): 
    parser.add_argument('--joined-dictionary', action='store_true', help='Generate joined dictionary')

    return parser





def main(args):

    print(args)

    os.makedirs(args.destdir, exist_ok=True)

tests/test_binaries.py

+7 −1
Changes for tests/test_binaries.py: 7 added lines, 1 removed line.
Original line number Diff line number Diff line
@@ -11,12 +11,16 @@ import random 
import sys

import tempfile

import unittest

from unittest import mock



import torch



from fairseq import options



import preprocess, train, generate, interactive

import preprocess

import train

import generate

import interactive





class TestBinaries(unittest.TestCase):
@@ -82,6 +86,8 @@ class TestBinaries(unittest.TestCase): 
                '--no-progress-bar',

            ],

        )

        with mock.patch('train.torch.cuda.device_count') as device_count:

            device_count.return_value = 1

            train.main(train_args)



    def generate(self, data_dir):

train.py

+2 −0
Changes for train.py: 2 added lines, 0 removed lines.
Original line number Diff line number Diff line
@@ -14,6 +14,7 @@ from distributed_train import main as distributed_main 
from multiprocessing_train import main as multiprocessing_main

from singleprocess_train import main as singleprocess_main





def main(args):

    if args.distributed_port > 0 \

            or args.distributed_init_method is not None:
@@ -23,6 +24,7 @@ def main(args): 
    else:

        singleprocess_main(args)





if __name__ == '__main__':

    parser = options.get_training_parser()

    args = options.parse_args_and_arch(parser)

fairseq/trainer.py

+7 −7

File changed.

Contains only whitespace changes.