Update tests.py

Smartappli · May 25, 2024 · 45e061b · 45e061b
1 parent 73c13b5
commit 45e061b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/fl_common/tests/tests.py b/fl_common/tests/tests.py
@@ -51,7 +51,8 @@ def test_data_augmentation_transform(self):
         self.assertTrue(
             any(
                 isinstance(t, transforms.Resize)
-                for t in transform.transforms)
+                for t in transform.transforms
+            ),
         )
         self.assertTrue(
             any(
@@ -99,7 +100,7 @@ def test_data_augmentation_transform(self):
             any(
                 isinstance(t, transforms.ToTensor)
                 for t in transform.transforms
-            )
+            ),
         )
         self.assertTrue(
             any(