Skip to content

Commit d324040

Browse files
Google-ML-Automationjax authors
authored andcommitted
Avoid "min() arg is an empty sequence" error after enabling "jax_explain_cache_misses".
PiperOrigin-RevId: 641381432
1 parent 57826d8 commit d324040

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

jax/_src/pjit.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1098,7 +1098,9 @@ def unpack(key):
10981098
f" {', '.join(map(repr, kwarg_keys))}")
10991099
dont_match = [set(t[1].node_data()[1]) for t in args_kwargs_trees # type: ignore
11001100
if t != [args_tree, kwargs_tree]]
1101-
close_kwargs = min(dont_match, key=set(kwarg_keys).symmetric_difference)
1101+
close_kwargs = min(
1102+
dont_match, key=set(kwarg_keys).symmetric_difference, default=None
1103+
)
11021104
if not close_kwargs:
11031105
p(" closest seen is passing no keyword args")
11041106
else:

0 commit comments

Comments
 (0)