Code formatting & cleanup for beam_search.py and inference.py - Githubissues

awslabs / sockeye

Sequence-to-sequence framework with a focus on Neural Machine Translation based on PyTorch

https://awslabs.github.io/sockeye/

Apache License 2.0

1.21k stars 323 forks source link

Code formatting & cleanup for beam_search.py and inference.py #1028

Closed fhieber closed 2 years ago

fhieber commented 2 years ago

reformatted a few lines to better adhere to max line length
refactored utils.gen_prefix_masking to simplify logic in beam search
rename numpy import from onp to np.

Pull Request Checklist

[x] Changes are complete (if posting work-in-progress code, prefix your pull request title with '[WIP]' until you can check this box.
[x] Unit tests pass (pytest)
[x] Passed code style checking (./style-check.sh)

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.