Skip to content

Commit 25a7e00

Browse files
committed
reduce test coverage overlap for weight formats
(onnx models should only have onnx weights as they should pass tests in any env wo torch/tf)
1 parent f49e639 commit 25a7e00

File tree

1 file changed

+2
-13
lines changed

1 file changed

+2
-13
lines changed

tests/conftest.py

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -115,23 +115,12 @@
115115
"unet2d_nuclei_broad_model",
116116
]
117117
)
118-
ONNX_MODELS = (
119-
[]
120-
if onnxruntime is None
121-
else [
122-
"hpa_densenet",
123-
"unet2d_multi_tensor",
124-
"unet2d_nuclei_broad_model",
125-
]
126-
)
118+
ONNX_MODELS = [] if onnxruntime is None else ["hpa_densenet"]
127119
TENSORFLOW_MODELS = (
128120
[]
129121
if tensorflow is None
130122
else (
131-
[
132-
"hpa_densenet",
133-
"stardist",
134-
]
123+
["stardist"]
135124
if tf_major_version == 1
136125
else [
137126
"unet2d_keras_tf2",

0 commit comments

Comments
 (0)