Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>

tests/kernels/core/test_rotary_embedding.py

@@ -4,8 +4,6 @@
 Tests for miscellaneous utilities
 """
 
-from typing import Optional
-
 import pytest
 import torch
 
@@ -17,7 +15,7 @@ def rotary_embedding_opcheck(
     rot,
     positions: torch.Tensor,
     query: torch.Tensor,
-    key: Optional[torch.Tensor] = None,
+    key: torch.Tensor | None = None,
 ):
     cos_sin_cache = rot.cos_sin_cache.to(query.device, dtype=query.dtype)