def list_info_files(dir_path: type_utils.PathLike) -> List[str]:
"""Returns name of info files within dir_path."""
path = os.fspath(dir_path)
return [
fname for fname in tf.io.gfile.listdir(path)
if not tf.io.gfile.isdir(os.path.join(path, fname)) and
not file_adapters.is_example_file(fname)
]
def list_info_files(dir_path: epath.PathLike) -> List[str]:
"""Returns name of info files within dir_path."""
from tensorflow_datasets.core import file_adapters # pylint: disable=g-import-not-at-top # pytype: disable=import-error
path = os.fspath(dir_path)
return [
fname for fname in tf.io.gfile.listdir(path)
if not tf.io.gfile.isdir(os.path.join(path, fname)) and
not file_adapters.is_example_file(fname)
]
def test_is_example_file():
assert file_adapters.is_example_file('example1.tfrecord')
assert file_adapters.is_example_file('example1.riegeli')
assert file_adapters.is_example_file('example1.tfrecord-00000-of-00001')
assert not file_adapters.is_example_file('example1.info')
