diff --git a/tests/tape_tests/test_feature_extraction.py b/tests/tape_tests/test_feature_extraction.py index 9cf70d18..73aaa3b5 100644 --- a/tests/tape_tests/test_feature_extraction.py +++ b/tests/tape_tests/test_feature_extraction.py @@ -84,7 +84,7 @@ def test_otsu_with_ensemble_all_bands(dask_client): result = ens.batch( licu.OtsuSplit(), band_to_calc=None, - ) + ).compute() assert result.shape == (2, 4) assert_array_equal(