diff --git a/tests/test_optimizer.py b/tests/test_optimizer.py index ae5c1dcd..752b8494 100644 --- a/tests/test_optimizer.py +++ b/tests/test_optimizer.py @@ -26,7 +26,6 @@ from torchspec.training.optimizer import BF16Optimizer - pytestmark = pytest.mark.skipif( not torch.cuda.is_available(), reason="fused AdamW found_inf handling requires CUDA",