diff --git a/tests/tests_fabric/accelerators/test_xla.py b/tests/tests_fabric/accelerators/test_xla.py index 0fcf6b45219be..762a3752c7e3a 100644 --- a/tests/tests_fabric/accelerators/test_xla.py +++ b/tests/tests_fabric/accelerators/test_xla.py @@ -14,7 +14,7 @@ import pytest -from lightning.fabric.accelerators.tpu import _XLA_AVAILABLE, XLAAccelerator +from lightning.fabric.accelerators.xla import _XLA_AVAILABLE, XLAAccelerator from tests_fabric.helpers.runif import RunIf