use code_format.sh to format all files (#64)

* CLN: auto format

Author: facaiy

configure.sh

@@ -13,6 +13,19 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
+# Usage: configure.sh [--quiet]
+#
+# Options:
+#  --quiet  Give less output.
+
+QUIET_FLAG=""
+if [[ $1 == "--quiet" ]]; then
+    QUIET_FLAG="--quiet"
+elif [[ ! -z "$1" ]]; then
+    echo "Found unsupported args: $@"
+    exit 1
+fi
+
 function write_to_bazelrc() {
   echo "$1" >> .bazelrc
 }
@@ -22,7 +35,7 @@ function write_action_env_to_bazelrc() {
 }
 
 [[ -f .bazelrc ]] && rm .bazelrc
-pip install -r requirements.txt
+pip install $QUIET_FLAG -r requirements.txt
 
 TF_CFLAGS=( $(python -c 'import tensorflow as tf; print(" ".join(tf.sysconfig.get_compile_flags()))') )
 TF_LFLAGS=( $(python -c 'import tensorflow as tf; print(" ".join(tf.sysconfig.get_link_flags()))') )

setup.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""Setup for pip package"""
+"""Setup for pip package."""
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
@@ -26,8 +26,9 @@
 version = {}
 base_dir = os.path.dirname(os.path.abspath(__file__))
 with open(os.path.join(base_dir, "tensorflow_addons", "version.py")) as fp:
+    # yapf: disable
     exec(fp.read(), version)
-
+    # yapf: enable
 
 REQUIRED_PACKAGES = [
     'six >= 1.10.0',

tensorflow_addons/BUILD

@@ -5,7 +5,7 @@ package(default_visibility = ["//visibility:public"])
 py_library(
     name = "tensorflow_addons",
     data = [
-            "__init__.py",
-            "version.py",
+        "__init__.py",
+        "version.py",
     ],
 )

tensorflow_addons/__init__.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""Useful extra functionality for TensorFlow maintained by SIG-addons"""
+"""Useful extra functionality for TensorFlow maintained by SIG-addons."""
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
@@ -22,13 +22,12 @@
 # tensorflow_probability namespace. Hence, we disable this lint check throughout
 # the file.
 #
-# pylint: disable=g-import-not-at-top
 
 
 # Ensure TensorFlow is importable and its version is sufficiently recent. This
 # needs to happen before anything else, since the imports below will try to
 # import tensorflow, too.
-def _ensure_tf_install():  # pylint: disable=g-statement-before-imports
+def _ensure_tf_install():
     """Attempt to import tensorflow, and ensure its version is sufficient.
 
     Raises:
@@ -60,8 +59,7 @@ def _ensure_tf_install():  # pylint: disable=g-statement-before-imports
             "This version of TensorFlow Addons requires TensorFlow "
             "version >= {required}; Detected an installation of version "
             "{present}. Please upgrade TensorFlow to proceed.".format(
-                required=required_tensorflow_version,
-                present=tf.__version__))
+                required=required_tensorflow_version, present=tf.__version__))
 
 
 _ensure_tf_install()
@@ -80,4 +78,3 @@ def _ensure_tf_install():  # pylint: disable=g-statement-before-imports
 from tensorflow_addons.custom_ops import text
 
 from tensorflow_addons.version import __version__
-# pylint: enable=g-import-not-at-top

tensorflow_addons/custom_ops/BUILD

@@ -7,7 +7,7 @@ py_library(
     srcs = ["__init__.py"],
     srcs_version = "PY2AND3",
     deps = [
-            "//tensorflow_addons/custom_ops/image:images_ops_py",
-            "//tensorflow_addons/custom_ops/text:text_py",
-        ]
+        "//tensorflow_addons/custom_ops/image:images_ops_py",
+        "//tensorflow_addons/custom_ops/text:text_py",
+    ],
 )

tensorflow_addons/custom_ops/image/BUILD

@@ -2,23 +2,25 @@ licenses(["notice"])  # Apache 2.0
 
 package(default_visibility = ["//visibility:public"])
 
-
 cc_binary(
     name = "python/_image_ops.so",
     srcs = [
         "cc/kernels/image_projective_transform_op.cc",
         "cc/kernels/image_projective_transform_op.h",
         "cc/ops/image_ops.cc",
     ],
+    copts = [
+        "-pthread",
+        "-std=c++11",
+        "-D_GLIBCXX_USE_CXX11_ABI=0",
+    ],
     linkshared = 1,
     deps = [
         "@local_config_tf//:libtensorflow_framework",
         "@local_config_tf//:tf_header_lib",
     ],
-    copts = ["-pthread", "-std=c++11", "-D_GLIBCXX_USE_CXX11_ABI=0"]
 )
 
-
 py_library(
     name = "images_ops_py",
     srcs = ([
@@ -27,23 +29,21 @@ py_library(
         "python/transform.py",
     ]),
     data = [
-        ":python/_image_ops.so"
+        ":python/_image_ops.so",
     ],
     srcs_version = "PY2AND3",
 )
 
-
 # TODO: use cuda_py_test later.
 py_test(
     name = "transform_ops_test",
     size = "small",
     srcs = [
-        "python/transform_test.py"
+        "python/transform_test.py",
     ],
     main = "python/transform_test.py",
+    srcs_version = "PY2AND3",
     deps = [
         ":images_ops_py",
     ],
-    srcs_version = "PY2AND3",
 )
-

tensorflow_addons/custom_ops/image/__init__.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""Image manipulation ops"""
+"""Image manipulation ops."""
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function

tensorflow_addons/custom_ops/image/cc/kernels/image_projective_transform_op.cc

@@ -19,11 +19,11 @@ limitations under the License.
 #define EIGEN_USE_GPU
 #endif  // GOOGLE_CUDA
 
+#include "tensorflow_addons/custom_ops/image/cc/kernels/image_projective_transform_op.h"
 #include "tensorflow/core/framework/op_kernel.h"
 #include "tensorflow/core/framework/register_types.h"
 #include "tensorflow/core/framework/types.h"
 #include "tensorflow/core/platform/types.h"
-#include "tensorflow_addons/custom_ops/image/cc/kernels/image_projective_transform_op.h"
 
 namespace tensorflow {
 
@@ -73,12 +73,11 @@ class ImageProjectiveTransform : public OpKernel {
     const Tensor& transform_t = ctx->input(1);
     OP_REQUIRES(ctx, images_t.shape().dims() == 4,
                 errors::InvalidArgument("Input images must have rank 4"));
-    OP_REQUIRES(ctx,
-                (TensorShapeUtils::IsMatrix(transform_t.shape()) &&
-                 (transform_t.dim_size(0) == images_t.dim_size(0) ||
-                  transform_t.dim_size(0) == 1) &&
-                 transform_t.dim_size(1) ==
-                     ProjectiveGenerator<Device, T>::kNumParameters),
+    OP_REQUIRES(ctx, (TensorShapeUtils::IsMatrix(transform_t.shape()) &&
+                      (transform_t.dim_size(0) == images_t.dim_size(0) ||
+                       transform_t.dim_size(0) == 1) &&
+                      transform_t.dim_size(1) ==
+                          ProjectiveGenerator<Device, T>::kNumParameters),
                 errors::InvalidArgument(
                     "Input transform should be num_images x 8 or 1 x 8"));
 
@@ -106,9 +105,8 @@ class ImageProjectiveTransform : public OpKernel {
 
     Tensor* output_t;
     OP_REQUIRES_OK(ctx, ctx->allocate_output(
-                            0,
-                            TensorShape({images_t.dim_size(0), out_height,
-                                         out_width, images_t.dim_size(3)}),
+                            0, TensorShape({images_t.dim_size(0), out_height,
+                                            out_width, images_t.dim_size(3)}),
                             &output_t));
     auto output = output_t->tensor<T, 4>();
     auto images = images_t.tensor<T, 4>();

tensorflow_addons/custom_ops/image/cc/kernels/image_projective_transform_op.h

@@ -45,9 +45,9 @@ class ProjectiveGenerator {
   static const int kNumParameters = 8;
 
   EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE
-  ProjectiveGenerator(typename TTypes<T, 4>::ConstTensor input,
-                      typename TTypes<float>::ConstMatrix transforms,
-                      const Interpolation interpolation)
+      ProjectiveGenerator(typename TTypes<T, 4>::ConstTensor input,
+                          typename TTypes<float>::ConstMatrix transforms,
+                          const Interpolation interpolation)
       : input_(input), transforms_(transforms), interpolation_(interpolation) {}
 
   EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE T

tensorflow_addons/custom_ops/image/cc/kernels/image_projective_transform_op_gpu.cu.cc

@@ -17,10 +17,10 @@ limitations under the License.
 
 #define EIGEN_USE_GPU
 
+#include "tensorflow_addons/custom_ops/image/cc/kernels/image_projective_transform_op.h"
 #include "tensorflow/core/framework/register_types.h"
 #include "tensorflow/core/framework/types.h"
 #include "tensorflow/core/platform/types.h"
-#include "tensorflow_addons/custom_ops/image/cc/kernels/image_projective_transform_op.h"
 
 namespace tensorflow {