From 7aa62810ed075a761ed63bbf515e5017b4d2d96d Mon Sep 17 00:00:00 2001 From: Xuhong Zhang Date: Wed, 19 Jun 2019 17:09:05 -0700 Subject: [PATCH] support ArrayType(StringType, _) for SequenceExample --- .../datasources/tfrecords/serde/DefaultTfRecordRowEncoder.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spark/spark-tensorflow-connector/src/main/scala/org/tensorflow/spark/datasources/tfrecords/serde/DefaultTfRecordRowEncoder.scala b/spark/spark-tensorflow-connector/src/main/scala/org/tensorflow/spark/datasources/tfrecords/serde/DefaultTfRecordRowEncoder.scala index bae47963..1c10a6b9 100644 --- a/spark/spark-tensorflow-connector/src/main/scala/org/tensorflow/spark/datasources/tfrecords/serde/DefaultTfRecordRowEncoder.scala +++ b/spark/spark-tensorflow-connector/src/main/scala/org/tensorflow/spark/datasources/tfrecords/serde/DefaultTfRecordRowEncoder.scala @@ -93,7 +93,7 @@ object DefaultTfRecordRowEncoder extends TfRecordRowEncoder { } } case (structField, index) => structField.dataType match { - case ArrayType(ArrayType(_, _), _) | ArrayType(StringType, _) => + case ArrayType(ArrayType(_, _), _) => val featureList = encodeFeatureList(row, structField, index) featureLists.putFeatureList(structField.name, featureList) case _ =>