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

Allow torch==1.10.2, rename unit test file #1021

Closed fhieber closed 2 years ago

fhieber commented 2 years ago

Previous pip install sockeye would uninstall torch==1.10.2 and install torch==1.10.0. We can safely depend on torch>=1.10.0 (1.10 and later).

Also rename a unit test module that was missed in the previous update.

Pull Request Checklist

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