Skip to content

Commit 86d482d

Browse files
committed
lint
1 parent e4d574a commit 86d482d

File tree

1 file changed

+2
-3
lines changed
  • onnxscript/function_libs/torch_lib/ops

1 file changed

+2
-3
lines changed

onnxscript/function_libs/torch_lib/ops/core.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4386,7 +4386,7 @@ def aten_index_put(
43864386
if (
43874387
len(indices) > 1
43884388
and any(
4389-
isinstance(indice, torch.onnx._internal.exporter._tensors.SymbolicTensor)
4389+
isinstance(indice, torch.onnx._internal.exporter._tensors.SymbolicTensor) # pylint: disable=protected-access
43904390
for indice in indices
43914391
)
43924392
and len(values.shape) == 1
@@ -4493,14 +4493,13 @@ def _make_range_or_cast(ind, shape_x, static_shape: bool, dim: int):
44934493
True,
44944494
)
44954495

4496-
rk1s = [(ind is None or len(ind.shape) == 1) for ind in indices]
44974496
shape_x = op.Shape(x)
44984497
exped = []
44994498
fixed = []
45004499
reshape_value_shape2 = []
45014500
expand_value_shape = []
45024501
for i, ind in enumerate(indices):
4503-
if isinstance(ind, torch.onnx._internal.exporter._tensors.SymbolicTensor):
4502+
if isinstance(ind, torch.onnx._internal.exporter._tensors.SymbolicTensor): # pylint: disable=protected-access
45044503
ind.dtype = ir.DataType.INT64
45054504
ind, expanded = _make_range_or_cast(ind, shape_x, False, i)
45064505
if expanded:

0 commit comments

Comments
 (0)