Fixing python3 issue for sparse_ops_test

v1vekkumar · Oct 30, 2016 · 20592a4 · 20592a4
1 parent 5e71c51
commit 20592a4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow/contrib/layers/python/ops/sparse_ops_test.py b/tensorflow/contrib/layers/python/ops/sparse_ops_test.py
@@ -67,7 +67,7 @@ def test_dense_to_sparse_tensor_1d_str(self):
     self.assertEqual(result.values.dtype, np.object)
     self.assertEqual(result.shape.dtype, np.int64)
     self.assertAllEqual([[0], [2]], result.indices)
-    self.assertAllEqual(['qwe', 'ewq'], result.values)
+    self.assertAllEqual([b'qwe', b'ewq'], result.values)
     self.assertAllEqual([4], result.shape)
 
   def test_dense_to_sparse_tensor_1d_str_special_ignore(self):
@@ -79,7 +79,7 @@ def test_dense_to_sparse_tensor_1d_str_special_ignore(self):
     self.assertEqual(result.values.dtype, np.object)
     self.assertEqual(result.shape.dtype, np.int64)
     self.assertAllEqual([[1], [2], [3]], result.indices)
-    self.assertAllEqual(['', 'ewq', ''], result.values)
+    self.assertAllEqual([b'', b'ewq', b''], result.values)
     self.assertAllEqual([4], result.shape)
 
   def test_dense_to_sparse_tensor_2d(self):