Fix typo in error message.

Google-ML-Automation · Google-ML-Automation · commit d69086db3529 · 2025-06-08T09:48:28.000-07:00
PiperOrigin-RevId: 768815050
diff --git a/jax/_src/pjit.py b/jax/_src/pjit.py
@@ -2993,7 +2993,7 @@ def reshard(xs, out_shardings):
     if ds is None:
       raise ValueError(
           'Reshard should only be used with out_shardings which are non-None '
-          'and have a nonempty mesh. Got sharding {s}.'
+          f'and have a nonempty mesh. Got sharding {s}.'
       )
     ds = ds.with_spec(ds.spec._normalized_spec_for_aval(x_aval.ndim))  # pytype: disable=attribute-error
     out_flat.append(reshard_p.bind(x, dst_sharding=ds))

Original file line number	Diff line number	Diff line change
`@@ -2993,7 +2993,7 @@ def reshard(xs, out_shardings):`
`2993`	`2993`	`if ds is None:`
`2994`	`2994`	`raise ValueError(`
`2995`	`2995`	`'Reshard should only be used with out_shardings which are non-None '`
`2996`		`- 'and have a nonempty mesh. Got sharding {s}.'`
	`2996`	`+ f'and have a nonempty mesh. Got sharding {s}.'`
`2997`	`2997`	`)`
`2998`	`2998`	`ds = ds.with_spec(ds.spec._normalized_spec_for_aval(x_aval.ndim)) # pytype: disable=attribute-error`
`2999`	`2999`	`out_flat.append(reshard_p.bind(x, dst_sharding=ds))`