Automated rollback of change 141490813

Change: 141521124

1 files changed, 5 insertions, 6 deletions

diff --git a/tensorflow/tools/dist_test/python/census_widendeep.py b/tensorflow/tools/dist_test/python/census_widendeep.py
index 62ff7f8f2f..309366e467 100644
--- a/tensorflow/tools/dist_test/python/census_widendeep.py
+++ b/tensorflow/tools/dist_test/python/census_widendeep.py
@@ -217,12 +217,11 @@ class CensusDataSource(object):
                        for k in self.continuous_columns}
     # Creates a dictionary mapping from each categorical feature column name (k)
     # to the values of that column stored in a tf.SparseTensor.
-    categorical_cols = {
-        k: tf.SparseTensor(
-            indices=[[i, 0] for i in range(df[k].size)],
-            values=df[k].values,
-            dense_shape=[df[k].size, 1])
-        for k in self.categorical_columns}
+    categorical_cols = {k: tf.SparseTensor(
+        indices=[[i, 0] for i in range(df[k].size)],
+        values=df[k].values,
+        shape=[df[k].size, 1])
+                        for k in self.categorical_columns}
     # Merges the two dictionaries into one.
     feature_cols = dict(continuous_cols.items() + categorical_cols.items())
     # Converts the label column into a constant Tensor.