diff --git a/tests/test_ops/test_roi_pool.py b/tests/test_ops/test_roi_pool.py index 43a88ae7c..f6d38d5af 100644 --- a/tests/test_ops/test_roi_pool.py +++ b/tests/test_ops/test_roi_pool.py @@ -102,9 +102,8 @@ class TestRoiPool: torch.double, marks=pytest.mark.skipif( IS_MLU_AVAILABLE or IS_NPU_AVAILABLE or IS_MUSA_AVAILABLE, - reason= - 'MLU, NPU, MUSA does not support for 64-bit floating point')), - torch.half + reason='MLU, NPU, MUSA ' + 'does not support for 64-bit floating point')), torch.half ]) def test_roipool_allclose(self, device, dtype): self._test_roipool_allclose(device, dtype)