Skip to content

Commit d4a450b

Browse files
committed
tidy up
1 parent 521cf5e commit d4a450b

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tests/function_libs/torch_lib/e2e_ops_tests.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -521,7 +521,7 @@ def forward(self, x):
521521
_testing.assert_onnx_program(onnx_program)
522522

523523
def test_unbind_dim0(self):
524-
"""Test unbind along dimension 0 (pytorch/pytorch#168969)."""
524+
"""Test unbind along dimension 0"""
525525

526526
class UnbindModel(torch.nn.Module):
527527
def forward(self, x):
@@ -534,7 +534,7 @@ def forward(self, x):
534534
_testing.assert_onnx_program(onnx_program)
535535

536536
def test_unbind_dim1(self):
537-
"""Test unbind along dimension 1 (pytorch/pytorch#168969)."""
537+
"""Test unbind along dimension 1"""
538538

539539
class UnbindModel(torch.nn.Module):
540540
def forward(self, x):
@@ -547,7 +547,7 @@ def forward(self, x):
547547
_testing.assert_onnx_program(onnx_program)
548548

549549
def test_unbind_negative_dim(self):
550-
"""Test unbind with negative dimension (pytorch/pytorch#168969)."""
550+
"""Test unbind with negative dimension"""
551551

552552
class UnbindModel(torch.nn.Module):
553553
def forward(self, x):
@@ -560,7 +560,7 @@ def forward(self, x):
560560
_testing.assert_onnx_program(onnx_program)
561561

562562
def test_unbind_size_one(self):
563-
"""Test unbind with dimension of size 1 (pytorch/pytorch#168969)."""
563+
"""Test unbind with dimension of size 1"""
564564

565565
class UnbindModel(torch.nn.Module):
566566
def forward(self, x):
@@ -573,7 +573,7 @@ def forward(self, x):
573573
_testing.assert_onnx_program(onnx_program)
574574

575575
def test_unbind_with_lstm(self):
576-
"""Test unbind in LSTM context (pytorch/pytorch#168969)."""
576+
"""Test unbind in LSTM context"""
577577

578578
class LSTMDecoder(torch.nn.Module):
579579
def __init__(self):

0 commit comments

Comments
 (0)