Added missing cupy test mark and fixed cupy threshold (#329)

* Added missing cupy test mark * threshold_local: ensure sigma argument is passed as numpy array
dask · Aug 2, 2023 · 264bd1d · 264bd1d
1 parent fba6645
commit 264bd1d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/dask_image/ndfilters/_threshold.py b/dask_image/ndfilters/_threshold.py
@@ -80,7 +80,7 @@ def threshold_local(image, block_size, method='gaussian', offset=0,
                                                mode=mode, cval=cval)
     elif method == 'gaussian':
         if param is None:
-            sigma = (da.asarray(block_size, like=image._meta) - 1) / 6.0
+            sigma = (np.array(block_size).astype(float) - 1) / 6.0
         else:
             sigma = param
         thresh_image = _gaussian.gaussian_filter(image, sigma, mode=mode,

diff --git a/tests/test_dask_image/test_ndfilters/test_cupy_threshold.py b/tests/test_dask_image/test_ndfilters/test_cupy_threshold.py
@@ -10,6 +10,7 @@
 cupy = pytest.importorskip("cupy", minversion="5.0.0")
 
 
+@pytest.mark.cupy
 @pytest.mark.parametrize('block_size', [
     3,
     [3, 3],