Update WishartCholesky.infer_shapes typing (#2059)

Author: fehiepsi

numpyro/distributions/continuous.py

@@ -3226,9 +3226,9 @@ def variance(self) -> ArrayLike:
     @staticmethod
     def infer_shapes(
         concentration: tuple[int, ...] = (),
-        scale_matrix: Optional[Array] = None,
-        rate_matrix: Optional[Array] = None,
-        scale_tril: Optional[Array] = None,
+        scale_matrix: Optional[tuple[int, ...]] = None,
+        rate_matrix: Optional[tuple[int, ...]] = None,
+        scale_tril: Optional[tuple[int, ...]] = None,
     ):
         assert_one_of(
             scale_matrix=scale_matrix,