diff --git a/requirements/adjust_versions.py b/requirements/adjust_versions.py index 4d0bbb5c00..278f2c71c7 100644 --- a/requirements/adjust_versions.py +++ b/requirements/adjust_versions.py @@ -3,16 +3,18 @@ import re import sys from typing import Dict, Optional +# IMPORTANT: this list needs to be sorted in reverse VERSIONS = [ dict(torch="1.10.0", torchvision="0.11.*", torchtext=""), # nightly + dict(torch="1.9.1", torchvision="0.10.1", torchtext="0.10.1"), dict(torch="1.9.0", torchvision="0.10.0", torchtext="0.10.0"), + dict(torch="1.8.2", torchvision="0.9.1", torchtext="0.9.1"), dict(torch="1.8.1", torchvision="0.9.1", torchtext="0.9.1"), dict(torch="1.8.0", torchvision="0.9.0", torchtext="0.9.0"), dict(torch="1.7.1", torchvision="0.8.2", torchtext="0.8.1"), dict(torch="1.7.0", torchvision="0.8.1", torchtext="0.8.0"), dict(torch="1.6.0", torchvision="0.7.0", torchtext="0.7"), ] -VERSIONS.sort(key=lambda v: v["torch"], reverse=True) def find_latest(ver: str) -> Dict[str, str]: