Skip to content

Commit c9b144f

Browse files
kuttire42pytorchmergebot
authored andcommitted
Replace assertEqualIgnoreTypes from common_methods_invocations.py (pytorch#84076)
This addresses TODO:38095 . More details at pytorch#38095 Fixes #ISSUE_NUMBER Pull Request resolved: pytorch#84076 Approved by: https://github.com/kit1980
1 parent b8fe0ed commit c9b144f

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

torch/testing/_internal/common_methods_invocations.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -17244,16 +17244,14 @@ def _compare_trilu_indices(
1724417244
torch.triu_indices(row, col, offset, dtype=dtype, device=device))
1724517245

1724617246
else:
17247-
# TODO(#38095): Replace assertEqualIgnoreType. See issue #38095
17248-
self.assertEqualIgnoreType(
17247+
self.assertEqual(
1724917248
torch.ones(row, col, device='cpu')
17250-
.tril(offset).nonzero().to(dtype).transpose(0, 1),
17249+
.tril(offset).nonzero().to(dtype=dtype).transpose(0, 1),
1725117250
torch.tril_indices(row, col, offset, dtype=dtype, device=device))
1725217251

17253-
# TODO(#38095): Replace assertEqualIgnoreType. See issue #38095
17254-
self.assertEqualIgnoreType(
17252+
self.assertEqual(
1725517253
torch.ones(row, col, device='cpu')
17256-
.triu(offset).nonzero().to(dtype).transpose(0, 1),
17254+
.triu(offset).nonzero().to(dtype=dtype).transpose(0, 1),
1725717255
torch.triu_indices(row, col, offset, dtype=dtype, device=device))
1725817256

1725917257

0 commit comments

Comments
 (0)