pytorch · XuehaiPan · Apr 5, 2025 · Apr 5, 2025 · Apr 5, 2025 · Apr 5, 2025
diff --git a/torch/_C/_VariableFunctions.pyi.in b/torch/_C/_VariableFunctions.pyi.in
@@ -11,15 +11,22 @@ from typing import (
     Iterator,
     Literal,
     NamedTuple,
-    Optional,
     overload,
     Sequence,
     TypeVar,
-    Union,
 )
 
 import torch
-from torch import contiguous_format, Generator, inf, memory_format, strided, SymInt, Tensor
+from torch import (
+    contiguous_format,
+    Generator,
+    inf,
+    memory_format,
+    strided,
+    SymInt,
+    Tensor,
+)
+from torch._prims_common import DeviceLikeType
 from torch.types import (
     _bool,
     _complex,
@@ -34,8 +41,6 @@ from torch.types import (
     Number,
 )
 
-from torch._prims_common import DeviceLikeType
+${all_directive}
 
 ${function_hints}
-
-${all_directive}