diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-15 15:44:56 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-15 15:49:01 -0700 |
| commit | 1e75c69339da2fbf2c5c5fbeb891243badae7ff8 (patch) | |
| tree | 81268169bbff6836bfbbd4e9866a1374f597a624 /tensorflow/tools/test | |
| parent | 6c62e650252ab32f83637a8de6720e73ffeca226 (diff) | |
Automated g4 rollback of changelist 189231636
PiperOrigin-RevId: 189258641
Diffstat (limited to 'tensorflow/tools/test')
| -rw-r--r-- | tensorflow/tools/test/upload_test_benchmarks.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/tensorflow/tools/test/upload_test_benchmarks.py b/tensorflow/tools/test/upload_test_benchmarks.py index edd093510e..77cc9f75f7 100644 --- a/tensorflow/tools/test/upload_test_benchmarks.py +++ b/tensorflow/tools/test/upload_test_benchmarks.py @@ -88,7 +88,6 @@ import os import shutil from google.cloud import datastore -from six import text_type def is_real_file(dirpath, fname): @@ -151,7 +150,7 @@ def upload_benchmark_data(client, data): """ test_result = json.loads(data) - test_name = text_type(test_result["name"]) + test_name = unicode(test_result["name"]) start_time = datetime.datetime.utcfromtimestamp( float(test_result["startTime"])) batch = [] @@ -163,7 +162,7 @@ def upload_benchmark_data(client, data): t_val.update({ "test": test_name, "start": start_time, - "info": text_type(data) + "info": unicode(data) }) batch.append(t_val) @@ -171,7 +170,7 @@ def upload_benchmark_data(client, data): # the attribute to be fetched and displayed. The full entry information is # also stored as a non-indexed JSON blob. for ent in test_result["entries"].get("entry", []): - ent_name = text_type(ent["name"]) + ent_name = unicode(ent["name"]) e_key = client.key("Entry") e_val = datastore.Entity(e_key, exclude_from_indexes=["info"]) e_val.update({ @@ -179,7 +178,7 @@ def upload_benchmark_data(client, data): "start": start_time, "entry": ent_name, "timing": ent["wallTime"], - "info": text_type(json.dumps(ent)) + "info": unicode(json.dumps(ent)) }) batch.append(e_val) |