Skip to content

Commit f8bc7f4

Browse files
fix flake
1 parent 1faea3c commit f8bc7f4

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

onmt/modules/copy_generator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,8 @@ def __init__(self, criterion, generator, tgt_vocab, normalize_by_length,
186186
self.tgt_vocab = tgt_vocab
187187
self.normalize_by_length = normalize_by_length
188188

189-
def _make_shard_state(self, batch, output, enc_src, enc_tgt, range_, attns):
189+
def _make_shard_state(self, batch, output, enc_src, enc_tgt,
190+
range_, attns):
190191
"""See base class for args description."""
191192
if getattr(batch, "alignment", None) is None:
192193
raise AssertionError("using -copy_attn you need to pass in "

0 commit comments

Comments
 (0)