Skip to content

Commit ce798d7

Browse files
committed
autofix
Signed-off-by: R. Garcia-Dias <[email protected]>
1 parent 016d3b0 commit ce798d7

File tree

2 files changed

+3
-8
lines changed

2 files changed

+3
-8
lines changed

tests/networks/blocks/test_patchembedding.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -53,11 +53,7 @@
5353
(2, params["embed_dim"], *([img_size // params["patch_size"]]) * params["spatial_dims"]),
5454
]
5555
for params in dict_product(
56-
patch_size=[2],
57-
in_chans=[1, 4],
58-
embed_dim=[6, 12],
59-
norm_layer=[nn.LayerNorm],
60-
spatial_dims=[2, 3],
56+
patch_size=[2], in_chans=[1, 4], embed_dim=[6, 12], norm_layer=[nn.LayerNorm], spatial_dims=[2, 3]
6157
)
6258
]
6359

tests/transforms/test_gibbs_noise.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,8 @@
2626
_, has_torch_fft = optional_import("torch.fft", name="fftshift")
2727

2828
TEST_CASES = dict_product(shape=((128, 64), (64, 48, 80)), input_type=TEST_NDARRAYS if has_torch_fft else [np.array])
29-
TEST_CASES = [
30-
[p_dict["shape"], p_dict["input_type"]] for p_dict in TEST_CASES
31-
]
29+
TEST_CASES = [[p_dict["shape"], p_dict["input_type"]] for p_dict in TEST_CASES]
30+
3231

3332
class TestGibbsNoise(unittest.TestCase):
3433
def setUp(self):

0 commit comments

Comments
 (0)