diff options
author | Yan Facai (颜发才) <facai.yan@gmail.com> | 2018-08-18 18:28:09 +0800 |
---|---|---|
committer | Yan Facai (颜发才) <facai.yan@gmail.com> | 2018-08-18 18:35:41 +0800 |
commit | 8fbafe6c7e75e1d931eca7202ea3a4c5ac8fc2dd (patch) | |
tree | b86bc57ad76ce970eb8bca92c218031778d7ee84 /tensorflow/contrib/data/python/kernel_tests/batch_dataset_op_test.py | |
parent | 6e0f1120fd7a6df805a8b712d2d4a38042576b46 (diff) |
CLN: fix code style
Diffstat (limited to 'tensorflow/contrib/data/python/kernel_tests/batch_dataset_op_test.py')
-rw-r--r-- | tensorflow/contrib/data/python/kernel_tests/batch_dataset_op_test.py | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/tensorflow/contrib/data/python/kernel_tests/batch_dataset_op_test.py b/tensorflow/contrib/data/python/kernel_tests/batch_dataset_op_test.py index ebc5160408..9d8e955245 100644 --- a/tensorflow/contrib/data/python/kernel_tests/batch_dataset_op_test.py +++ b/tensorflow/contrib/data/python/kernel_tests/batch_dataset_op_test.py @@ -759,8 +759,8 @@ class RestructuredDatasetTest(test.TestCase): def create_unknown_shape_dataset(x): return script_ops.py_func( lambda _: ( # pylint: disable=g-long-lambda - np.ones(2, dtype=np.float32), - np.zeros((3, 4), dtype=np.int32)), + np.ones(2, dtype=np.float32), + np.zeros((3, 4), dtype=np.int32)), [x], [dtypes.float32, dtypes.int32]) @@ -789,8 +789,8 @@ class RestructuredDatasetTest(test.TestCase): def create_unknown_shape_dataset(x): return script_ops.py_func( lambda _: ( # pylint: disable=g-long-lambda - np.ones(2, dtype=np.float32), - np.zeros((3, 4), dtype=np.int32)), + np.ones(2, dtype=np.float32), + np.zeros((3, 4), dtype=np.int32)), [x], [dtypes.float32, dtypes.int32]) @@ -802,7 +802,7 @@ class RestructuredDatasetTest(test.TestCase): wrong_shapes = (tensor_shape.TensorShape(2), tensor_shape.TensorShape((3, 10))) iterator = ( - dataset.apply(batching.assert_element_shape(wrong_shapes)) + dataset.apply(batching.assert_element_shape(wrong_shapes)) .make_initializable_iterator()) init_op = iterator.initializer get_next = iterator.get_next() @@ -854,8 +854,8 @@ class RestructuredDatasetTest(test.TestCase): def create_unknown_shape_dataset(x): return script_ops.py_func( lambda _: ( # pylint: disable=g-long-lambda - np.ones(2, dtype=np.float32), - np.zeros((3, 4), dtype=np.int32)), + np.ones(2, dtype=np.float32), + np.zeros((3, 4), dtype=np.int32)), [x], [dtypes.float32, dtypes.int32]) @@ -884,8 +884,8 @@ class RestructuredDatasetTest(test.TestCase): def create_unknown_shape_dataset(x): return script_ops.py_func( lambda _: ( # pylint: disable=g-long-lambda - np.ones(2, dtype=np.float32), - np.zeros((3, 4), dtype=np.int32)), + np.ones(2, dtype=np.float32), + np.zeros((3, 4), dtype=np.int32)), [x], [dtypes.float32, dtypes.int32]) @@ -897,7 +897,7 @@ class RestructuredDatasetTest(test.TestCase): wrong_shapes = (tensor_shape.TensorShape(2), tensor_shape.TensorShape((None, 10))) iterator = ( - dataset.apply(batching.assert_element_shape(wrong_shapes)) + dataset.apply(batching.assert_element_shape(wrong_shapes)) .make_initializable_iterator()) init_op = iterator.initializer get_next = iterator.get_next() |