Formatting

build_deps/configure.py

@@ -16,13 +16,12 @@
 #
 
 
+import logging
 import os
 import pathlib
 import platform
-import logging
 
 import tensorflow as tf
-
 from packaging.version import Version
 
 _TFA_BAZELRC = ".bazelrc"

keras_cv/losses/__init__.py

@@ -17,4 +17,3 @@
 from keras_cv.losses.iou_loss import IoULoss
 from keras_cv.losses.simclr_loss import SimCLRLoss
 from keras_cv.losses.smooth_l1 import SmoothL1Loss
-from keras_cv.losses.smooth_l1 import SmoothL1Loss

keras_cv/losses/iou_3d_loss.py

@@ -18,5 +18,6 @@
 from tensorflow.python.platform import resource_loader
 
 iou_3d_ops = load_library.load_op_library(
-    resource_loader.get_path_to_datafile('../custom_ops/_zero_out_ops.so'))
+    resource_loader.get_path_to_datafile("../custom_ops/_zero_out_ops.so")
+)
 iou_3d = iou_3d_ops.zero_out

keras_cv/losses/iou_3d_loss_test.py

@@ -19,11 +19,9 @@
 
 
 class IoU3DTest(tf.test.TestCase):
+    def testOpCall(self):
+        self.assertAllClose(IoU3DLoss([[1, 2], [3, 4]]), [[1, 0], [0, 0]])
 
-  def testOpCall(self):
-      self.assertAllClose(
-          IoU3DLoss([[1, 2], [3, 4]]), [[1, 0], [0, 0]])
 
-
-if __name__ == '__main__':
-  test.main()
+if __name__ == "__main__":
+    tf.test.main()

setup.py

@@ -25,13 +25,13 @@
 
 
 class BinaryDistribution(Distribution):
-  """This class is needed in order to create OS specific wheels."""
+    """This class is needed in order to create OS specific wheels."""
 
-  def has_ext_modules(self):
-    return True
+    def has_ext_modules(self):
+        return True
 
-  def is_pure(self):
-    return False
+    def is_pure(self):
+        return False
 
 
 setup(