@@ -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