Autofix

Author: garciadias

tests/apps/detection/networks/test_retinanet.py

@@ -91,17 +91,11 @@
 MODEL_LIST = [resnet10, resnet18, resnet34, resnet50, resnet101, resnet152, resnet200]
 
 TEST_CASES = []
-for params in dict_product(
-    model=MODEL_LIST,
-    case=CASE_LIST,
-):
+for params in dict_product(model=MODEL_LIST, case=CASE_LIST):
     TEST_CASES.append([params["model"], *params["case"]])
 
 TEST_CASES_TS = []
-for params in dict_product(
-    model=MODEL_LIST,
-    case=[TEST_CASE_1],
-):
+for params in dict_product(model=MODEL_LIST, case=[TEST_CASE_1]):
     TEST_CASES_TS.append([params["model"], *params["case"]])
 
 

tests/apps/detection/test_box_transform.py

@@ -73,8 +73,7 @@
 for params in dict_product(ndarray_type=TEST_NDARRAYS):
     p = params["ndarray_type"]
     TESTS_2D.append(
-        [{"boxes": p(boxes_2d), "image": p(image_2d), "labels": p(labels_2d)},
-         p([[[0, 2], [0, 2]], [[1, 0], [0, 0]]])]
+        [{"boxes": p(boxes_2d), "image": p(image_2d), "labels": p(labels_2d)}, p([[[0, 2], [0, 2]], [[1, 0], [0, 0]]])]
     )
 
 # Use dict_product for TESTS_2D_mask

tests/data/meta_tensor/test_meta_tensor.py

@@ -38,10 +38,7 @@
 
 # Replace nested loops with dict_product
 TESTS = []
-for params in dict_product(
-    device=TEST_DEVICES,
-    dtype=DTYPES
-):
+for params in dict_product(device=TEST_DEVICES, dtype=DTYPES):
     TESTS.append((*params["device"], *params["dtype"]))  # type: ignore
 
 

tests/networks/blocks/test_patchembedding.py

@@ -57,12 +57,7 @@
 
 TEST_CASE_PATCHEMBED = []
 for params in dict_product(
-    patch_size=[2],
-    in_chans=[1, 4],
-    img_size=[96],
-    embed_dim=[6, 12],
-    norm_layer=[nn.LayerNorm],
-    nd=[2, 3],
+    patch_size=[2], in_chans=[1, 4], img_size=[96], embed_dim=[6, 12], norm_layer=[nn.LayerNorm], nd=[2, 3]
 ):
     test_case = [
         {