Skip to content

Commit bc11a63

Browse files
committed
Clarify jax.make_jaxpr docstring
1 parent bcd4048 commit bc11a63

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

jax/_src/api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2142,7 +2142,7 @@ def make_jaxpr(
21422142
return_shape: bool = False,
21432143
abstracted_axes: Any | None = None,
21442144
) -> Callable[..., core.ClosedJaxpr | tuple[core.ClosedJaxpr, Any]]:
2145-
"""Creates a function that produces its jaxpr given example args.
2145+
"""Create a function that returns the jaxpr of ``fun`` given example args.
21462146
21472147
Args:
21482148
fun: The function whose ``jaxpr`` is to be computed. Its positional

0 commit comments

Comments
 (0)