From 8008aa9ef251f708e9c74b0c7823fad8b78c553f Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Tue, 12 May 2020 11:09:02 -0400 Subject: [PATCH] Fix error message in optimizers piecewise_constant. (#3061) --- jax/experimental/optimizers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jax/experimental/optimizers.py b/jax/experimental/optimizers.py index c4164f4a71d5..20348d9269bb 100644 --- a/jax/experimental/optimizers.py +++ b/jax/experimental/optimizers.py @@ -499,7 +499,7 @@ def piecewise_constant(boundaries, values): if not boundaries.ndim == values.ndim == 1: raise ValueError("boundaries and values must be sequences") if not boundaries.shape[0] == values.shape[0] - 1: - raise ValueError("boundaries length must be one longer than values length") + raise ValueError("boundaries length must be one shorter than values length") def schedule(i): return values[jnp.sum(i > boundaries)]