use base_version to check torch version in torch_less_than_1_11 #16806
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
The current check
version.parse(torch.__version__) < version.parse("1.11")
doesn't work if the version has additional stuff after it like1.11.0a0+17540c5
.New way:
version.parse(version.parse(torch.__version__).base_version) < version.parse("1.11")
This change, as suggested by @whoknowsB here, fixes it.
Fixes #16587 and #14375
Who can review?
@LysandreJik
@sgugger