format.

keras_cv/layers/preprocessing3d/global_rotation.py

@@ -48,7 +48,9 @@ class GlobalRandomRotation(base_augmentation_layer_3d.BaseAugmentationLayer3D):
 
     """
 
-    def __init__(self, max_rotation_angle_x, max_rotation_angle_y, max_rotation_angle_z, **kwargs):
+    def __init__(
+        self, max_rotation_angle_x, max_rotation_angle_y, max_rotation_angle_z, **kwargs
+    ):
         super().__init__(**kwargs)
         if max_rotation_angle_x < 0:
             raise ValueError("max_rotation_angle_x must be >=0.")
@@ -70,7 +72,12 @@ def get_random_transformation(self, **kwargs):
         random_rotation_z = self._random_generator.random_uniform(
             (), minval=-self._max_rotation_angle_z, maxval=self._max_rotation_angle_z
         )
-        return {"pose": tf.stack([0, 0, 0, random_rotation_z, random_rotation_x, random_rotation_y], axis=0)}
+        return {
+            "pose": tf.stack(
+                [0, 0, 0, random_rotation_z, random_rotation_x, random_rotation_y],
+                axis=0,
+            )
+        }
 
     def augment_point_clouds_bounding_boxes(
         self, point_clouds, bounding_boxes, transformation, **kwargs

keras_cv/layers/preprocessing3d/global_rotation_test.py

@@ -23,31 +23,39 @@
 
 class GlobalRandomRotationTest(tf.test.TestCase):
     def test_augment_point_clouds_and_bounding_boxes(self):
-        add_layer = GlobalRandomRotation( max_rotation_angle_x=1.0, max_rotation_angle_y=1.0, max_rotation_angle_z=1.0)
+        add_layer = GlobalRandomRotation(
+            max_rotation_angle_x=1.0, max_rotation_angle_y=1.0, max_rotation_angle_z=1.0
+        )
         point_clouds = np.random.random(size=(2, 50, 10)).astype("float32")
         bounding_boxes = np.random.random(size=(2, 10, 7)).astype("float32")
         inputs = {POINT_CLOUDS: point_clouds, BOUNDING_BOXES: bounding_boxes}
         outputs = add_layer(inputs)
         self.assertNotAllClose(inputs, outputs)
 
     def test_not_augment_point_clouds_and_bounding_boxes(self):
-        add_layer = GlobalRandomRotation( max_rotation_angle_x=0.0, max_rotation_angle_y=0.0, max_rotation_angle_z=0.0)
+        add_layer = GlobalRandomRotation(
+            max_rotation_angle_x=0.0, max_rotation_angle_y=0.0, max_rotation_angle_z=0.0
+        )
         point_clouds = np.random.random(size=(2, 50, 10)).astype("float32")
         bounding_boxes = np.random.random(size=(2, 10, 7)).astype("float32")
         inputs = {POINT_CLOUDS: point_clouds, BOUNDING_BOXES: bounding_boxes}
         outputs = add_layer(inputs)
         self.assertAllClose(inputs, outputs)
 
     def test_augment_batch_point_clouds_and_bounding_boxes(self):
-        add_layer = GlobalRandomRotation( max_rotation_angle_x=1.0, max_rotation_angle_y=1.0, max_rotation_angle_z=1.0)
+        add_layer = GlobalRandomRotation(
+            max_rotation_angle_x=1.0, max_rotation_angle_y=1.0, max_rotation_angle_z=1.0
+        )
         point_clouds = np.random.random(size=(3, 2, 50, 10)).astype("float32")
         bounding_boxes = np.random.random(size=(3, 2, 10, 7)).astype("float32")
         inputs = {POINT_CLOUDS: point_clouds, BOUNDING_BOXES: bounding_boxes}
         outputs = add_layer(inputs)
         self.assertNotAllClose(inputs, outputs)
 
     def test_not_augment_batch_point_clouds_and_bounding_boxes(self):
-        add_layer = GlobalRandomRotation( max_rotation_angle_x=0.0, max_rotation_angle_y=0.0, max_rotation_angle_z=0.0)
+        add_layer = GlobalRandomRotation(
+            max_rotation_angle_x=0.0, max_rotation_angle_y=0.0, max_rotation_angle_z=0.0
+        )
         point_clouds = np.random.random(size=(3, 2, 50, 10)).astype("float32")
         bounding_boxes = np.random.random(size=(3, 2, 10, 7)).astype("float32")
         inputs = {POINT_CLOUDS: point_clouds, BOUNDING_BOXES: bounding_boxes}