Skip to content

Commit 5f54cfc

Browse files
authored
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
1 parent b908c50 commit 5f54cfc

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

setup.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -130,11 +130,11 @@ def _parse_requirements_file(file_path):
130130
"onnxruntime>=1.7.0",
131131
]
132132
_image_classification_deps = [
133-
"torchvision>=0.3.0,<=0.13",
133+
"torchvision>=0.3.0,<0.14",
134134
"opencv-python<=4.6.0.66",
135135
]
136136
_yolo_integration_deps = [
137-
"torchvision>=0.3.0,<=0.13",
137+
"torchvision>=0.3.0,<0.14",
138138
"opencv-python<=4.6.0.66",
139139
]
140140
_openpifpaf_integration_deps = [

tests/utils/test_engine_mocking.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,8 @@ def test_mock_engine_calls(engine_mock: MagicMock):
3838
engine_mock.assert_called_once_with(
3939
os.path.join(
4040
os.path.expanduser("~"),
41-
".cache/sparsezoo/84774c96-ab7d-4b3b-ab8c-2509d7bfcb09/model.onnx",
41+
".cache/sparsezoo/neuralmagic/",
42+
"resnet_v1-50-imagenet-pruned85.4block_quantized/model.onnx",
4243
),
4344
3,
4445
1,

0 commit comments

Comments
 (0)