fix mypy errora

Signed-off-by: R. Garcia-Dias <[email protected]>

Author: garciadias

tests/transforms/test_gibbs_noise.py

@@ -25,8 +25,9 @@
 
 _, has_torch_fft = optional_import("torch.fft", name="fftshift")
 
-TEST_CASES = dict_product(shape=((128, 64), (64, 48, 80)), input_type=TEST_NDARRAYS if has_torch_fft else [np.array])
-TEST_CASES = [[p_dict["shape"], p_dict["input_type"]] for p_dict in TEST_CASES]
+shapes = ((128, 64), (64, 48, 80))
+input_types = TEST_NDARRAYS if has_torch_fft else [np.array]
+TEST_CASES = [[p_dict["shape"], p_dict["input_type"]] for p_dict in dict_product(shape=shapes, input_type=input_types)]
 
 
 class TestGibbsNoise(unittest.TestCase):