fix bug

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
open-mmlab · Sep 30, 2022 · ea1d8f8 · ea1d8f8
1 parent 4ce938c
commit ea1d8f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_ops/test_focal_loss.py b/tests/test_ops/test_focal_loss.py
@@ -150,7 +150,7 @@ def test_sigmoid_float(self, device):
         pytest.param(
             'npu',
             marks=pytest.mark.skipif(
-                not IS_CUDA_AVAILABLE, reason='requires NPU support')),
+                not IS_NPU_AVAILABLE, reason='requires NPU support')),
         pytest.param(
             'cuda',
             marks=pytest.mark.skipif(