Black

talmolab · Mar 29, 2024 · 628d2ab · 628d2ab
1 parent 0a8fdc0
commit 628d2ab
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/tests/nn/data/test_augmentation.py b/tests/nn/data/test_augmentation.py
@@ -45,16 +45,15 @@ def augmentation_config(rotation_min_angle, rotation_max_angle):
     return augmentation.AugmentationConfig(
         rotate=True,
         rotation_min_angle=rotation_min_angle,
-        rotation_max_angle=rotation_max_angle
+        rotation_max_angle=rotation_max_angle,
     )
 
 
 @pytest.fixture
 def dummy_dataset(dummy_image_data, dummy_instances_data_zeros):
-    dataset = tf.data.Dataset.from_tensor_slices({
-        "image": [dummy_image_data],
-        "instances": [dummy_instances_data_zeros]
-    })
+    dataset = tf.data.Dataset.from_tensor_slices(
+        {"image": [dummy_image_data], "instances": [dummy_instances_data_zeros]}
+    )
     return dataset
 
 
@@ -168,7 +167,6 @@ def test_augmentation_edges(min_labels):
     assert example["instances"].shape == (3, 2, 2)
 
 
-
 def test_random_cropper(min_labels):
     cropper = augmentation.RandomCropper(crop_height=64, crop_width=32)
     assert "image" in cropper.input_keys