We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 57826d8 commit d324040Copy full SHA for d324040
jax/_src/pjit.py
@@ -1098,7 +1098,9 @@ def unpack(key):
1098
f" {', '.join(map(repr, kwarg_keys))}")
1099
dont_match = [set(t[1].node_data()[1]) for t in args_kwargs_trees # type: ignore
1100
if t != [args_tree, kwargs_tree]]
1101
- close_kwargs = min(dont_match, key=set(kwarg_keys).symmetric_difference)
+ close_kwargs = min(
1102
+ dont_match, key=set(kwarg_keys).symmetric_difference, default=None
1103
+ )
1104
if not close_kwargs:
1105
p(" closest seen is passing no keyword args")
1106
else:
0 commit comments