Skip to content

Commit 1e3339d

Browse files
garciadiasericspod
andauthored
Apply suggestions from code review
Co-authored-by: Eric Kerfoot <[email protected]> Signed-off-by: Rafael Garcia-Dias <[email protected]>
1 parent 76033a6 commit 1e3339d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/transforms/test_gibbs_noise.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727

2828
shapes = ((128, 64), (64, 48, 80))
2929
input_types = TEST_NDARRAYS if has_torch_fft else [np.array]
30-
TEST_CASES = [[p_dict["shape"], p_dict["input_type"]] for p_dict in dict_product(shape=shapes, input_type=input_types)]
30+
TEST_CASES = list(map(list, product(shape=shapes, input_type=input_types)))
3131

3232

3333
class TestGibbsNoise(unittest.TestCase):

0 commit comments

Comments
 (0)