Fix tuple type hint to a variable length tuple.

pyro-ppl · Mar 17, 2024 · c4e4fb9 · c4e4fb9
1 parent b812fe4
commit c4e4fb9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyro/ops/stats.py b/pyro/ops/stats.py
@@ -264,7 +264,7 @@ def quantile(input, probs, dim=0):
 
 def weighed_quantile(
     input: torch.Tensor,
-    probs: Union[List[float], Tuple[float], torch.Tensor],
+    probs: Union[List[float], Tuple[float, ...], torch.Tensor],
     log_weights: torch.Tensor,
     dim: int = 0,
 ) -> torch.Tensor: