🎨 Format Python code with psf/black

fl_common/models/hiera.py

@@ -22,8 +22,8 @@ def get_hiera_model(hiera_type, num_classes):
         "hiera_base_plus_224",
         "hiera_large_224",
         "hiera_huge_224",
-        'hiera_small_abswin_256',
-        'hiera_base_abswin_256',
+        "hiera_small_abswin_256",
+        "hiera_base_abswin_256",
     }
 
     if hiera_type not in valid_hiera_types:

fl_common/models/segmentation/snp.py

@@ -577,10 +577,16 @@ def test_model_combinations(encoders, architectures):
                         in_channels=3,
                         classes=2,
                     )
-                    preprocess_input = get_preprocessing_fn(encoder["model"], pretrained='imagenet')
-                    print(f'Success: Created {arch["model"]} with {encoder["model"]}')
+                    preprocess_input = get_preprocessing_fn(
+                        encoder["model"], pretrained="imagenet"
+                    )
+                    print(
+                        f'Success: Created {arch["model"]} with {encoder["model"]}'
+                    )
                 except Exception as e:
-                    print(f'Failed: {arch["model"]} with {encoder["model"]} - {e}')
+                    print(
+                        f'Failed: {arch["model"]} with {encoder["model"]} - {e}'
+                    )
 
 
 test_model_combinations(encoders, architectures)

fl_server/tests/tests_server.py

@@ -76,10 +76,11 @@ def test_launch_node_main(self):
         )
         self.client_humani.upload_dataset(dataset)
 
-        ds_client = login('node_humani',
-                          "[email protected]",
-                          "abc123",
-                          )
+        ds_client = login(
+            "node_humani",
+            "[email protected]",
+            "abc123",
+        )
 
         asset = ds_client.datasets[-1].assets["ages"]
         mock = asset.mock