Require torchvision<0.14 rather than torchvision<=0.13 (#1034)

* Require `torchvision<0.14` rather than `torchvision<=0.13` * Update test_engine_mocking.py * Update test_engine_mocking.py
neuralmagic · May 23, 2023 · 5f54cfc · 5f54cfc
1 parent b908c50
commit 5f54cfc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -130,11 +130,11 @@ def _parse_requirements_file(file_path):
     "onnxruntime>=1.7.0",
 ]
 _image_classification_deps = [
-    "torchvision>=0.3.0,<=0.13",
+    "torchvision>=0.3.0,<0.14",
     "opencv-python<=4.6.0.66",
 ]
 _yolo_integration_deps = [
-    "torchvision>=0.3.0,<=0.13",
+    "torchvision>=0.3.0,<0.14",
     "opencv-python<=4.6.0.66",
 ]
 _openpifpaf_integration_deps = [

diff --git a/tests/utils/test_engine_mocking.py b/tests/utils/test_engine_mocking.py
@@ -38,7 +38,8 @@ def test_mock_engine_calls(engine_mock: MagicMock):
     engine_mock.assert_called_once_with(
         os.path.join(
             os.path.expanduser("~"),
-            ".cache/sparsezoo/84774c96-ab7d-4b3b-ab8c-2509d7bfcb09/model.onnx",
+            ".cache/sparsezoo/neuralmagic/",
+            "resnet_v1-50-imagenet-pruned85.4block_quantized/model.onnx",
         ),
         3,
         1,