autofix

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

Author: garciadias

tests/networks/blocks/test_CABlock.py

@@ -36,12 +36,7 @@
         (2, params["dim"], *([16] * params["spatial_dims"])),
         (2, params["dim"], *([16] * params["spatial_dims"])),
     ]
-    for params in dict_product(
-        spatial_dims=[2, 3],
-        dim=[32, 64, 128],
-        num_heads=[2, 4, 8],
-        bias=[True, False],
-    )
+    for params in dict_product(spatial_dims=[2, 3], dim=[32, 64, 128], num_heads=[2, 4, 8], bias=[True, False])
 ]
 
 

tests/networks/blocks/test_crossattention.py

@@ -40,13 +40,14 @@
         (2, 512, params["hidden_size"]),
     ]
     for params in dict_product(
-    dropout_rate=np.linspace(0, 1, 4),
-    hidden_size=[360, 480, 600, 768],
-    num_heads=[4, 6, 8, 12],
-    rel_pos_embedding_val=[None, RelPosEmbedding.DECOMPOSED],
-    input_size=[(16, 32), (8, 8, 8)],
-    flash_attn=[True, False],
-)]
+        dropout_rate=np.linspace(0, 1, 4),
+        hidden_size=[360, 480, 600, 768],
+        num_heads=[4, 6, 8, 12],
+        rel_pos_embedding_val=[None, RelPosEmbedding.DECOMPOSED],
+        input_size=[(16, 32), (8, 8, 8)],
+        flash_attn=[True, False],
+    )
+]
 
 
 class TestResBlock(unittest.TestCase):

tests/test_masked_autoencoder_vit.py

@@ -18,7 +18,7 @@
 
 from monai.networks import eval_mode
 from monai.networks.nets.masked_autoencoder_vit import MaskedAutoEncoderViT
-from tests.test_utils import skip_if_quick, dict_product
+from tests.test_utils import dict_product, skip_if_quick
 
 TEST_CASE_MaskedAutoEncoderViT = []
 
@@ -62,11 +62,7 @@
         if nd == 2:
             model_params["spatial_dims"] = 2
 
-        test_case = [
-            model_params,
-            input_shape,
-            expected_shape,
-        ]
+        test_case = [model_params, input_shape, expected_shape]
         TEST_CASE_MaskedAutoEncoderViT.append(test_case)
 
 TEST_CASE_ill_args = [