File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -41,7 +41,7 @@ def test_predict_with_pipeline(prep: Prep):
41
41
assert out == prep .output_sample
42
42
43
43
44
- @pytest .mark .parameterize ("tensor_input" , ["numpy" , "xarray" ])
44
+ @pytest .mark .parametrize ("tensor_input" , ["numpy" , "xarray" ])
45
45
def test_predict_with_model_description (
46
46
tensor_input : Literal ["numpy" , "xarray" ], prep : Prep
47
47
):
@@ -64,7 +64,7 @@ def test_predict_with_model_description(
64
64
assert out == prep .output_sample
65
65
66
66
67
- @pytest .mark .parameterize ("with_proces" , [True , False ])
67
+ @pytest .mark .parametrize ("with_proces" , [True , False ])
68
68
def test_predict_with_blocking (with_procs : bool , prep : Prep ):
69
69
input_block_shape : Mapping [MemberId , Mapping [AxisId , int ]] = {
70
70
list (prep .input_sample .members )[0 ]: {
You can’t perform that action at this time.
0 commit comments