File tree 2 files changed +4
-3
lines changed
2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -130,11 +130,11 @@ def _parse_requirements_file(file_path):
130
130
"onnxruntime>=1.7.0" ,
131
131
]
132
132
_image_classification_deps = [
133
- "torchvision>=0.3.0,<=0.13 " ,
133
+ "torchvision>=0.3.0,<0.14 " ,
134
134
"opencv-python<=4.6.0.66" ,
135
135
]
136
136
_yolo_integration_deps = [
137
- "torchvision>=0.3.0,<=0.13 " ,
137
+ "torchvision>=0.3.0,<0.14 " ,
138
138
"opencv-python<=4.6.0.66" ,
139
139
]
140
140
_openpifpaf_integration_deps = [
Original file line number Diff line number Diff line change @@ -38,7 +38,8 @@ def test_mock_engine_calls(engine_mock: MagicMock):
38
38
engine_mock .assert_called_once_with (
39
39
os .path .join (
40
40
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" ,
42
43
),
43
44
3 ,
44
45
1 ,
You can’t perform that action at this time.
0 commit comments