Added cuda decorator

ornlneutronimaging · Jul 3, 2024 · 9d38233 · 9d38233
1 parent 0e9fa09
commit 9d38233
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/bm3dornl/test_bm3d_ring_artifact_removal_ms.py b/tests/unit/bm3dornl/test_bm3d_ring_artifact_removal_ms.py
@@ -11,7 +11,7 @@
 def setup_sinogram():
     return np.random.rand(size_x, size_y)
 
-
+@pytest.mark.cuda_required
 def test_bm3d_ring_artifact_removal_ms(
     setup_sinogram,
 ):