Skip to content

Commit 7d303fe

Browse files
committed
autofix
1 parent 88cf0b5 commit 7d303fe

2 files changed

Lines changed: 6 additions & 11 deletions

File tree

tests/networks/nets/test_mednext.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,13 +23,12 @@
2323
device = "cuda" if torch.cuda.is_available() else "cpu"
2424

2525
TEST_CASE_MEDNEXT = [
26-
[
27-
params,
28-
(2, 1, *([16] * params["spatial_dims"])),
29-
(2, 2, *([16] * params["spatial_dims"])),
30-
]
26+
[params, (2, 1, *([16] * params["spatial_dims"])), (2, 2, *([16] * params["spatial_dims"]))]
3127
for params in dict_product(
32-
spatial_dims=range(2, 4), init_filters=[8, 16], deep_supervision=[False, True], use_residual_connection=[False, True]
28+
spatial_dims=range(2, 4),
29+
init_filters=[8, 16],
30+
deep_supervision=[False, True],
31+
use_residual_connection=[False, True],
3332
)
3433
]
3534
TEST_CASE_MEDNEXT_2 = [

tests/networks/nets/test_segresnet.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,7 @@
3939
]
4040

4141
TEST_CASE_SEGRESNET_2 = [
42-
[
43-
params,
44-
(2, 1, *([16] * params["spatial_dims"])),
45-
(2, params["out_channels"], *([16] * params["spatial_dims"])),
46-
]
42+
[params, (2, 1, *([16] * params["spatial_dims"])), (2, params["out_channels"], *([16] * params["spatial_dims"]))]
4743
for params in dict_product(
4844
spatial_dims=range(2, 4), init_filters=[8, 16], out_channels=range(1, 3), upsample_mode=list(UpsampleMode)
4945
)

0 commit comments

Comments
 (0)