Commit cb0d7b2a authored by Louis Martin's avatar Louis Martin Committed by Myle Ott
Browse files

Fix flake8 warnings

parent 7333d04d
Loading
Loading
Loading
Loading
+2 −0
Changes for fairseq/__init__.py: 2 added lines, 0 removed lines.
Original line number Diff line number Diff line
@@ -7,3 +7,5 @@
#

from .multiprocessing_pdb import pdb

__all__ = ['pdb']
+1 −1
Changes for fairseq/bleu.py: 1 added line, 1 removed line.
Original line number Diff line number Diff line
@@ -97,7 +97,7 @@ class Scorer(object):
    def result_string(self, order=4):
        assert order <= 4, "BLEU scores for order > 4 aren't supported"
        fmt = 'BLEU{} = {:2.2f}, {:2.1f}'
        for i in range(1, order):
        for _ in range(1, order):
            fmt += '/{:2.1f}'
        fmt += ' (BP={:.3f}, ratio={:.3f}, syslen={}, reflen={})'
        bleup = [p * 100 for p in self.precision()[:order]]
+1 −1
Changes for fairseq/data.py: 1 added line, 1 removed line.
Original line number Diff line number Diff line
@@ -326,7 +326,7 @@ def shuffled_batches_by_size(src, dst, max_tokens=None, epoch=1, sample=0, max_p

        batches = result
    else:
        for i in range(epoch - 1):
        for _ in range(epoch - 1):
            np.random.shuffle(batches)

    return batches
+1 −1
Changes for fairseq/sequence_generator.py: 1 added line, 1 removed line.
Original line number Diff line number Diff line
@@ -126,7 +126,7 @@ class SequenceGenerator(object):

        # helper function for allocating buffers on the fly
        buffers = {}
        def buffer(name, type_of=tokens):
        def buffer(name, type_of=tokens):  # noqa
            if name not in buffers:
                buffers[name] = type_of.new()
            return buffers[name]
+1 −1
Changes for fairseq/utils.py: 1 added line, 1 removed line.
Original line number Diff line number Diff line
@@ -41,7 +41,7 @@ def torch_persistent_save(*args, **kwargs):
    for i in range(3):
        try:
            return torch.save(*args, **kwargs)
        except:
        except Exception:
            if i == 2:
                logging.error(traceback.format_exc())

Loading