fix import

huggingface · Dec 24, 2024 · 20f5e3c · 20f5e3c
1 parent 51f651b
commit 20f5e3c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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

diff --git 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