diff options
author | Yong Tang <yong.tang.github@outlook.com> | 2018-06-23 17:17:22 +0000 |
---|---|---|
committer | Yong Tang <yong.tang.github@outlook.com> | 2018-06-23 19:37:32 +0000 |
commit | a1548df10159c0485ac83477af91b31ffe2d5d26 (patch) | |
tree | 09bd23202d91d3ae2006ed042eb28f7b52914d22 /tensorflow/contrib/image | |
parent | e32b8e1d505a9b7dcfb70707a77c830271a27fcf (diff) |
Pylint fix
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'tensorflow/contrib/image')
-rw-r--r-- | tensorflow/contrib/image/python/kernel_tests/image_ops_test.py | 3 | ||||
-rw-r--r-- | tensorflow/contrib/image/python/ops/image_ops.py | 3 |
2 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/contrib/image/python/kernel_tests/image_ops_test.py b/tensorflow/contrib/image/python/kernel_tests/image_ops_test.py index 1bc7adb856..88a5c9f079 100644 --- a/tensorflow/contrib/image/python/kernel_tests/image_ops_test.py +++ b/tensorflow/contrib/image/python/kernel_tests/image_ops_test.py @@ -30,7 +30,8 @@ from tensorflow.python.ops import math_ops from tensorflow.python.platform import googletest _DTYPES = set( - [dtypes.uint8, dtypes.int32, dtypes.int64, dtypes.float16, dtypes.float32, dtypes.float64]) + [dtypes.uint8, dtypes.int32, dtypes.int64, + dtypes.float16, dtypes.float32, dtypes.float64]) class ImageOpsTest(test_util.TensorFlowTestCase): diff --git a/tensorflow/contrib/image/python/ops/image_ops.py b/tensorflow/contrib/image/python/ops/image_ops.py index ab739799e7..86b0ffe9a0 100644 --- a/tensorflow/contrib/image/python/ops/image_ops.py +++ b/tensorflow/contrib/image/python/ops/image_ops.py @@ -33,7 +33,8 @@ _image_ops_so = loader.load_op_library( resource_loader.get_path_to_datafile("_image_ops.so")) _IMAGE_DTYPES = set( - [dtypes.uint8, dtypes.int32, dtypes.int64, dtypes.float16, dtypes.float32, dtypes.float64]) + [dtypes.uint8, dtypes.int32, dtypes.int64, + dtypes.float16, dtypes.float32, dtypes.float64]) ops.RegisterShape("ImageConnectedComponents")(common_shapes.call_cpp_shape_fn) ops.RegisterShape("ImageProjectiveTransform")(common_shapes.call_cpp_shape_fn) |