Skip to content

Commit d01b61e

Browse files
committed
fix formatting issues for three more files
Signed-off-by: Han Wang <freddie.wanah@gmail.com>
1 parent 00b1465 commit d01b61e

File tree

3 files changed

+6
-42
lines changed

3 files changed

+6
-42
lines changed

tests/test_hausdorff_loss.py

Lines changed: 2 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -219,13 +219,7 @@ def test_ill_opts(self):
219219
with self.assertRaisesRegex(ValueError, ""):
220220
HausdorffDTLoss(reduction=None)(chn_input, chn_target)
221221

222-
@parameterized.expand(
223-
[
224-
(False, False, False),
225-
(False, True, False),
226-
(False, False, True),
227-
]
228-
)
222+
@parameterized.expand([(False, False, False), (False, True, False), (False, False, True)])
229223
def test_input_warnings(self, include_background, softmax, to_onehot_y):
230224
chn_input = torch.ones((1, 1, 1, 3))
231225
chn_target = torch.ones((1, 1, 1, 3))
@@ -257,13 +251,7 @@ def test_ill_opts(self):
257251
with self.assertRaisesRegex(ValueError, ""):
258252
LogHausdorffDTLoss(reduction=None)(chn_input, chn_target)
259253

260-
@parameterized.expand(
261-
[
262-
(False, False, False),
263-
(False, True, False),
264-
(False, False, True),
265-
]
266-
)
254+
@parameterized.expand([(False, False, False), (False, True, False), (False, False, True)])
267255
def test_input_warnings(self, include_background, softmax, to_onehot_y):
268256
chn_input = torch.ones((1, 1, 1, 3))
269257
chn_target = torch.ones((1, 1, 1, 3))

tests/test_tversky_loss.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -165,13 +165,7 @@ def test_ill_shape(self):
165165
with self.assertRaisesRegex(ValueError, ""):
166166
TverskyLoss(reduction=None)(chn_input, chn_target)
167167

168-
@parameterized.expand(
169-
[
170-
(False, False, False),
171-
(False, True, False),
172-
(False, False, True),
173-
]
174-
)
168+
@parameterized.expand([(False, False, False), (False, True, False), (False, False, True)])
175169
def test_input_warnings(self, include_background, softmax, to_onehot_y):
176170
chn_input = torch.ones((1, 1, 3))
177171
chn_target = torch.ones((1, 1, 3))

tests/test_ultrasound_confidence_map_transform.py

Lines changed: 3 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -537,12 +537,7 @@ def test_parameters(self):
537537
UltrasoundConfidenceMapTransform(sink_mode="unknown")
538538

539539
@parameterized.expand(
540-
[
541-
("all", SINK_ALL_OUTPUT),
542-
("mid", SINK_MID_OUTPUT),
543-
("min", SINK_MIN_OUTPUT),
544-
("mask", SINK_MASK_OUTPUT, True),
545-
]
540+
[("all", SINK_ALL_OUTPUT), ("mid", SINK_MID_OUTPUT), ("min", SINK_MIN_OUTPUT), ("mask", SINK_MASK_OUTPUT, True)]
546541
)
547542
def test_ultrasound_confidence_map_transform(self, sink_mode, expected_output, use_mask=False):
548543
# RGB image
@@ -589,14 +584,7 @@ def test_multi_channel_2d(self, sink_mode, expected_output, use_mask=False):
589584
self.assertIsInstance(result_np, np.ndarray)
590585
assert_allclose(result_np, expected_output, rtol=1e-4, atol=1e-4)
591586

592-
@parameterized.expand(
593-
[
594-
("all",),
595-
("mid",),
596-
("min",),
597-
("mask",),
598-
]
599-
)
587+
@parameterized.expand([("all",), ("mid",), ("min",), ("mask",)])
600588
def test_non_one_first_dim(self, sink_mode):
601589
transform = UltrasoundConfidenceMapTransform(sink_mode=sink_mode)
602590
input_img_rgb = np.repeat(self.input_img_np, 3, axis=0)
@@ -631,13 +619,7 @@ def test_no_first_dim(self, sink_mode):
631619
with self.assertRaises(ValueError):
632620
transform(input_img_rgb, self.input_mask_np)
633621

634-
@parameterized.expand(
635-
[
636-
("all",),
637-
("mid",),
638-
("min",),
639-
]
640-
)
622+
@parameterized.expand([("all",), ("mid",), ("min",)])
641623
def test_sink_mode(self, mode):
642624
transform = UltrasoundConfidenceMapTransform(sink_mode=mode)
643625

0 commit comments

Comments
 (0)