allow patch versions of torchvision 0.14 beyond .0 (#1556)

bfineran · web-flow · commit 2f86f086b68d · 2023-05-10T17:15:22.000-04:00
as written, only 0.14.0 would be allowed
diff --git a/setup.py b/setup.py
@@ -66,11 +66,10 @@
     "gputils",
 ]
 _pytorch_all_deps = _pytorch_deps + [
-    "torchvision>=0.3.0,<=0.14",
+    "torchvision>=0.3.0,<0.15",
     "torchaudio<=0.13",
-    "torchvision>=0.3.0,<=0.14",
 ]
-_pytorch_vision_deps = _pytorch_deps + ["torchvision>=0.3.0,<=0.14"]
+_pytorch_vision_deps = _pytorch_deps + ["torchvision>=0.3.0,<0.15"]
 _transformers_deps = _pytorch_deps + [
     f"{'nm-transformers' if is_release else 'nm-transformers-nightly'}"
     f"~={version_nm_deps}",

Original file line number	Diff line number	Diff line change
`@@ -66,11 +66,10 @@`
`66`	`66`	`"gputils",`
`67`	`67`	`]`
`68`	`68`	`_pytorch_all_deps = _pytorch_deps + [`
`69`		`- "torchvision>=0.3.0,<=0.14",`
	`69`	`+ "torchvision>=0.3.0,<0.15",`
`70`	`70`	`"torchaudio<=0.13",`
`71`		`- "torchvision>=0.3.0,<=0.14",`
`72`	`71`	`]`
`73`		`-_pytorch_vision_deps = _pytorch_deps + ["torchvision>=0.3.0,<=0.14"]`
	`72`	`+_pytorch_vision_deps = _pytorch_deps + ["torchvision>=0.3.0,<0.15"]`
`74`	`73`	`_transformers_deps = _pytorch_deps + [`
`75`	`74`	`f"{'nm-transformers' if is_release else 'nm-transformers-nightly'}"`
`76`	`75`	`f"~={version_nm_deps}",`