tensorflow / tensorflow/datasets

Better progression bar for extraction

Open
#2,979 0 comments 2 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

contributions welcome enhancement
Dominant language
Python
Stars
4.6k
Forks
1.6k
Avg merge
3h 54m
Merged PRs (30d)
1

Description

Currently, extracting big archive made it looks TFDS is frozen. It would be great to have a more granular progression bar (tqdm) counting the extract number of files, rather than just extracted/non-extracted.

https://github.com/tensorflow/datasets/blob/0540c979cba848d814d8266e807352063b4ef355/tensorflow_datasets/core/download/extractor.py#L61

For the implementation, it would be like the download bar which indicates the combined size of all download: https://github.com/tensorflow/datasets/blob/0540c979cba848d814d8266e807352063b4ef355/tensorflow_datasets/core/download/downloader.py#L138

To detect the number of file, we could use TarFile.getnames(), ZipFile.namelist()

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Read tensorflow_datasets/core/download/extractor.py at the linked extraction logic, then compare its progress handling with tensorflow_datasets/core/download/downloader.py at the linked download bar. Use TarFile.getnames() and ZipFile.namelist() as suggested to determine archive file counts; done means extraction reports granular tqdm progress rather than only extracted or not extracted.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
tooling
Issue type
Feature
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Clearly specified
Newbie friendliness
55/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.