Skip to content

Commit 85bb863

Browse files
committed
Fix 1.14 build failure
Signed-off-by: Yong Tang <[email protected]>
1 parent 363c42d commit 85bb863

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/test_archive_eager.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ def test_gz():
3333
filename = "file://" + filename
3434

3535
format, entries = archive_io.read_archive_entries(filename, ["gz", "tar.gz"]) # pylint: disable=redefined-builtin
36-
assert format.numpy() == "gz"
36+
assert format.numpy().decode() == "gz"
3737
assert entries.shape == [1]
3838

3939
elements = archive_io.read_archive(filename, format, entries)
@@ -56,7 +56,7 @@ def test_none():
5656
filename = "file://" + filename
5757

5858
format, entries = archive_io.read_archive_entries(filename, ["none", "gz"]) # pylint: disable=redefined-builtin
59-
assert format.numpy() == "none"
59+
assert format.numpy().decode() == "none"
6060
assert entries.shape == [1]
6161

6262
elements = archive_io.read_archive(filename, format, entries)
@@ -73,10 +73,10 @@ def test_tar_gz():
7373
filename = "file://" + filename
7474

7575
format, entries = archive_io.read_archive_entries(filename, ["gz", "tar.gz"]) # pylint: disable=redefined-builtin
76-
assert format.numpy() == "tar.gz"
76+
assert format.numpy().decode() == "tar.gz"
7777
assert entries.shape == [2]
78-
assert entries[0].numpy() == "parquet_cpp_example.parquet.1"
79-
assert entries[1].numpy() == "parquet_cpp_example.parquet.2"
78+
assert entries[0].numpy().decode() == "parquet_cpp_example.parquet.1"
79+
assert entries[1].numpy().decode() == "parquet_cpp_example.parquet.2"
8080

8181
elements = archive_io.read_archive(filename, format, entries)
8282
assert elements.shape == [2]
@@ -103,7 +103,7 @@ def test_dataset():
103103
lambda f: () + (f,) + archive_io.read_archive_entries(f, "tar.gz")
104104
).map(
105105
lambda f, format, e: (tf.broadcast_to(f, tf.shape(e)), tf.broadcast_to(format, tf.shape(e)), e)
106-
).unbatch().map(archive_io.read_archive)
106+
).apply(tf.data.experimental.unbatch()).map(archive_io.read_archive)
107107

108108
expected_filename = os.path.join(
109109
os.path.dirname(os.path.abspath(__file__)),

0 commit comments

Comments
 (0)