Skip to content

Commit a115fa9

Browse files
committed
fixme
Signed-off-by: Justin Chu <[email protected]>
1 parent 4d9ca4d commit a115fa9

File tree

1 file changed

+1
-1
lines changed
  • onnxscript/function_libs/torch_lib/ops

1 file changed

+1
-1
lines changed

onnxscript/function_libs/torch_lib/ops/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4812,7 +4812,7 @@ def _aten_index_bool(self: TensorType, indices: Sequence[Optional[BOOL]]) -> Ten
48124812
trans_perm = trans_perm[-1:] + trans_perm[:-1]
48134813
for _ in range(count_of_none):
48144814
result = op.Transpose(result, perm=trans_perm)
4815-
# TODO(justinchuby): Even though this logic passes the tests, it still looks strange:
4815+
# FIXME(justinchuby): Even though this logic passes the tests, it still looks strange:
48164816
# why does it return early here instead of continuing to process the remaining indices?
48174817
# I think the assumption here is that there can be only one Boolean index in the indices list?
48184818
return result

0 commit comments

Comments
 (0)