fix queue type hints using simplequeue

Author: bhimrazy

src/lightning/fabric/strategies/launchers/xla.py

@@ -17,6 +17,7 @@
 from typing import TYPE_CHECKING, Any, Union
 
 import torch.multiprocessing as mp
+from torch.multiprocessing.queue import SimpleQueue
 from typing_extensions import override
 
 from lightning.fabric.accelerators.xla import _XLA_AVAILABLE
@@ -69,7 +70,7 @@ def launch(self, function: Callable, *args: Any, **kwargs: Any) -> Any:
             **kwargs: Optional keyword arguments to be passed to the given function.
 
         """
-        return_queue: queue.Queue | mp.SimpleQueue
+        return_queue: queue.Queue | SimpleQueue
         return_queue = mp.Manager().Queue()
 
         import torch_xla.distributed.xla_multiprocessing as xmp
@@ -97,7 +98,7 @@ def _wrapping_function(
         function: Callable,
         args: Any,
         kwargs: Any,
-        return_queue: mp.SimpleQueue | queue.Queue,
+        return_queue: queue.Queue | SimpleQueue,
         global_states: _GlobalStateSnapshot | None = None,
     ) -> None:
         import torch_xla.core.xla_model as xm