Fix style errors

This commit is contained in:
drebain 2020-08-25 15:31:41 -07:00
parent b857488021
commit 9225122330
2 changed files with 2 additions and 2 deletions
tensorflow/python/kernel_tests

View File

@ -2048,10 +2048,10 @@ cuda_py_test(
name = "dynamic_partition_op_test",
size = "medium",
srcs = ["dynamic_partition_op_test.py"],
tfrt_enabled = True,
tags = [
"multi_and_single_gpu",
],
tfrt_enabled = True,
deps = [
"//tensorflow/python:array_ops",
"//tensorflow/python:client_testlib",

View File

@ -354,7 +354,7 @@ class DynamicPartitionTest(test.TestCase):
results = []
for device in device_list:
with ops.device(device.name):
data = constant_op.constant(np.zeros((1000, )))
data = constant_op.constant(np.zeros((1000,)))
partitions = constant_op.constant(np.arange(1000, dtype=np.int32) % 10)
result = data_flow_ops.dynamic_partition(data, partitions, 10)
results.append(self.evaluate(result))