Fix import warnings (#15144)

2025-12-14 21:25:18 -08:00
parent 9e9a61691e
commit 702426b06a
2 changed files with 2 additions and 2 deletions
@@ -36,7 +36,7 @@ def tensor_torch2ms(x: torch.Tensor):
    return ms_tensor


-def tensor_ms2torch(x: ms.Tensor):
+def tensor_ms2torch(x: "ms.Tensor"):
    if x is None or not isinstance(x, ms.Tensor):
        return x