Make type check for quantile stricter to match what is actually tested.
Make type check for quantile stricter to match what is actually tested.

google/jax,google/jax,tensorflow/probability,google/jax,google/jax,tensorflow/probability