We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 294eca3 commit 7192035Copy full SHA for 7192035
1 file changed
tests/function_libs/torch_lib/ops_test_data.py
@@ -189,8 +189,8 @@ def _embedding_bag_input_wrangler(
189
args: list[Any], kwargs: dict[str, Any]
190
) -> tuple[list[Any], dict[str, Any]]:
191
# ONNX attributes cannot be None; omit padding_idx if it's None.
192
- padding_idx = kwargs.pop("padding_idx", "___MISSING___")
193
- if padding_idx != "___MISSING___":
+ if "padding_idx" in kwargs:
+ padding_idx = kwargs.pop("padding_idx")
194
if padding_idx is not None:
195
kwargs["padding_idx"] = int(padding_idx)
196
0 commit comments