diff --git a/tests/test_kwargs_handlers.py b/tests/test_kwargs_handlers.py index bf8bd1bd939..d0415dab4bb 100644 --- a/tests/test_kwargs_handlers.py +++ b/tests/test_kwargs_handlers.py @@ -27,7 +27,6 @@ path_in_accelerate_package, require_multi_device, require_non_cpu, - require_non_xpu, ) from accelerate.test_utils.testing import slow from accelerate.utils import AutocastKwargs, KwargsHandler, ProfileKwargs, TorchDynamoPlugin, clear_environment @@ -50,7 +49,6 @@ def test_kwargs_handler(self): assert MockClass(a=2, c=2.25).to_kwargs() == {"a": 2, "c": 2.25} @require_non_cpu - @require_non_xpu def test_grad_scaler_kwargs(self): # If no defaults are changed, `to_kwargs` returns an empty dict. scaler_handler = GradScalerKwargs(init_scale=1024, growth_factor=2) diff --git a/tests/test_optimizer.py b/tests/test_optimizer.py index 374733ae762..4f74f5ded44 100644 --- a/tests/test_optimizer.py +++ b/tests/test_optimizer.py @@ -19,7 +19,7 @@ from accelerate import Accelerator from accelerate.state import AcceleratorState -from accelerate.test_utils import require_cpu, require_non_cpu, require_non_xpu +from accelerate.test_utils import require_cpu, require_non_cpu @require_cpu @@ -37,7 +37,6 @@ def test_accelerated_optimizer_pickling(self): @require_non_cpu -@require_non_xpu class OptimizerTester(unittest.TestCase): def test_accelerated_optimizer_step_was_skipped(self): model = torch.nn.Linear(5, 5)