wishart.py: use Tensor instead of torch.Tensor annotation

pytorch · randolf-scholz · Jan 2, 2025 · Jan 2, 2025 · Jan 2, 2025 · Jan 2, 2025
commit 64d9269011d11f82179859b3b997b86ea703679c
diff --git a/torch/distributions/wishart.py b/torch/distributions/wishart.py
@@ -76,10 +76,10 @@ class Wishart(ExponentialFamily):
 
     def __init__(
         self,
-        df: Union[torch.Tensor, Number],
-        covariance_matrix: Optional[torch.Tensor] = None,
-        precision_matrix: Optional[torch.Tensor] = None,
-        scale_tril: Optional[torch.Tensor] = None,
+        df: Union[Tensor, Number],
+        covariance_matrix: Optional[Tensor] = None,
+        precision_matrix: Optional[Tensor] = None,
+        scale_tril: Optional[Tensor] = None,
         validate_args=None,
     ):
         assert (covariance_matrix is not None) + (scale_tril is not None) + (