Skip to content

Commit d2da96e

Browse files
committed
style: fixed linting issues in code
1 parent 6e41cfe commit d2da96e

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

onnxscript/function_libs/torch_lib/ops/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3115,7 +3115,7 @@ def aten_embedding_bag_padding_idx(
31153115

31163116
if padding_idx is not None:
31173117
# Call the existing function for handling padding_idx
3118-
result, offset2bag, bag_size, max_indices =_aten_embedding_bag_1d_padding_idx_onnx(
3118+
result, offset2bag, bag_size, max_indices = _aten_embedding_bag_1d_padding_idx_onnx(
31193119
weight,
31203120
indices,
31213121
offsets,

tests/function_libs/torch_lib/ops_test_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,6 +203,7 @@ def _embedding_bag_input_wrangler(
203203

204204
return args, kwargs
205205

206+
206207
def _amin_amax_input_wrangler(
207208
args: list[Any], kwargs: dict[str, Any]
208209
) -> tuple[list[Any], dict[str, Any]]:
@@ -1068,7 +1069,6 @@ def _where_input_wrangler(
10681069
core_ops.aten_embedding_bag,
10691070
input_wrangler=_embedding_bag_input_wrangler,
10701071
),
1071-
10721072
TorchLibOpInfo(
10731073
"ops.aten.embedding_bag.padding_idx",
10741074
core_ops.aten_embedding_bag_padding_idx,

0 commit comments

Comments
 (0)