Skip to content

Commit f407fcc

Browse files
authored
Skip HoVerNet Loss for pytorch 1.9.1 and older (#5401)
1 parent 9ded7cf commit f407fcc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_hovernet_loss.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ def test_shape_generator(num_classes=1, num_objects=3, batch_size=1, height=5, w
169169
]
170170

171171

172-
@SkipIfBeforePyTorchVersion((1, 9))
172+
@SkipIfBeforePyTorchVersion((1, 10))
173173
class TestHoverNetLoss(unittest.TestCase):
174174
@parameterized.expand(CASES)
175175
def test_shape(self, input_param, expected_loss):

0 commit comments

Comments
 (0)