@@ -110,9 +110,7 @@ def check_result():
110110 check_result ()
111111
112112 # test with fixed padding
113- predict_image (
114- model , in_path , out_path , padding = {"x" : original_shape [0 ], "y" : original_shape [1 ], "mode" : "fixed" }
115- )
113+ predict_image (model , in_path , out_path , padding = {"x" : original_shape [0 ], "y" : original_shape [1 ], "mode" : "fixed" })
116114 check_result ()
117115
118116 # test with automated padding
@@ -135,7 +133,7 @@ def test_predict_image_with_padding_channel_last(stardist, tmp_path):
135133 _test_predict_with_padding (stardist , tmp_path )
136134
137135
138- def _test_predict_image_with_tiling (model , tmp_path , exp_mean_deviation ):
136+ def _test_predict_image_with_tiling (model , tmp_path : Path , exp_mean_deviation ):
139137 from bioimageio .core .prediction import predict_image
140138
141139 spec = load_resource_description (model )
@@ -168,27 +166,27 @@ def check_result():
168166
169167# prediction with tiling with the parameters above may not be suited for any model
170168# so we only run it for the pytorch unet2d here
171- def test_predict_image_with_tiling_1 (unet2d_nuclei_broad_model , tmp_path ):
169+ def test_predict_image_with_tiling_1 (unet2d_nuclei_broad_model , tmp_path : Path ):
172170 _test_predict_image_with_tiling (unet2d_nuclei_broad_model , tmp_path , 0.012 )
173171
174172
175- def test_predict_image_with_tiling_2 (unet2d_diff_output_shape , tmp_path ):
176- _test_predict_image_with_tiling (unet2d_diff_output_shape , tmp_path , 0.012 )
173+ def test_predict_image_with_tiling_2 (unet2d_diff_output_shape , tmp_path : Path ):
174+ _test_predict_image_with_tiling (unet2d_diff_output_shape , tmp_path , 0.06 )
177175
178176
179- def test_predict_image_with_tiling_3 (shape_change_model , tmp_path ):
177+ def test_predict_image_with_tiling_3 (shape_change_model , tmp_path : Path ):
180178 _test_predict_image_with_tiling (shape_change_model , tmp_path , 0.012 )
181179
182180
183- def test_predict_image_with_tiling_channel_last (stardist , tmp_path ):
181+ def test_predict_image_with_tiling_channel_last (stardist , tmp_path : Path ):
184182 _test_predict_image_with_tiling (stardist , tmp_path , 0.13 )
185183
186184
187- def test_predict_image_with_tiling_fixed_output_shape (unet2d_fixed_shape , tmp_path ):
185+ def test_predict_image_with_tiling_fixed_output_shape (unet2d_fixed_shape , tmp_path : Path ):
188186 _test_predict_image_with_tiling (unet2d_fixed_shape , tmp_path , 0.025 )
189187
190188
191- def test_predict_images (unet2d_nuclei_broad_model , tmp_path ):
189+ def test_predict_images (unet2d_nuclei_broad_model , tmp_path : Path ):
192190 from bioimageio .core .prediction import predict_images
193191
194192 n_images = 5
0 commit comments