Skip to content

Commit 7192035

Browse files
committed
fix: clean up _embedding_bag_input_wrangler padding_idx check
1 parent 294eca3 commit 7192035

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/function_libs/torch_lib/ops_test_data.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -189,8 +189,8 @@ def _embedding_bag_input_wrangler(
189189
args: list[Any], kwargs: dict[str, Any]
190190
) -> tuple[list[Any], dict[str, Any]]:
191191
# 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___":
192+
if "padding_idx" in kwargs:
193+
padding_idx = kwargs.pop("padding_idx")
194194
if padding_idx is not None:
195195
kwargs["padding_idx"] = int(padding_idx)
196196

0 commit comments

Comments
 (0)