Fix tests

from torch.autograd import Variable

from torch.serialization import default_restore_location

from fairseq import criterions, data, models, progress_bar, tokenizer

from fairseq import criterions, progress_bar, tokenizer

def parse_args_and_arch(parser):

    from fairseq import models

    args = parser.parse_args()

    args.model = models.arch_model_map[args.arch]

    args = getattr(models, args.model).parse_arch(args)

def build_model(args, src_dict, dst_dict):

    from fairseq import models

    assert hasattr(models, args.model), 'Missing model type'

    return getattr(models, args.model).build_model(args, src_dict, dst_dict)

    The source and target dictionaries can be given explicitly, or loaded from

    the `data_dir` directory.

    """

    from fairseq import data

    # load model architectures and weights

    states = []

    for filename in filenames: