Fix smoke tests for GCS to be Python3 compatible.

PiperOrigin-RevId: 303024321
Change-Id: I83626c22b2110cb0b79c4f020cbc13451811f4f3
This commit is contained in:
A. Unique TensorFlower 2020-03-25 19:41:25 -07:00 committed by TensorFlower Gardener
parent c94d33c7d3
commit 81415e2bfc

View File

@ -44,7 +44,7 @@ def create_examples(num_examples, input_mean):
examples = []
for row in range(num_examples):
ex = example_pb2.Example()
ex.features.feature["id"].bytes_list.value.append(str(ids[row, 0]))
ex.features.feature["id"].bytes_list.value.append(bytes(ids[row, 0]))
ex.features.feature["target"].float_list.value.append(target[row, 0])
ex.features.feature["inputs"].float_list.value.append(inputs[row, 0])
examples.append(ex)