Skip to content

Add tqdm progress bar to initial data generation#1

Open
JPBotelho wants to merge 1 commit intoAxiomMath:mainfrom
JPBotelho:tqdm-data-generation
Open

Add tqdm progress bar to initial data generation#1
JPBotelho wants to merge 1 commit intoAxiomMath:mainfrom
JPBotelho:tqdm-data-generation

Conversation

@JPBotelho
Copy link
Copy Markdown

What?

Adds a tqdm progress bar to the data generation step.

Why?

Added this progress bar in my fork after checking on a job after 10h and seeing it stuck on data generation.

"Testing"

Ran axplorer locally, verified the progress bar showed up.
image

image

tqdm is already defined as a dependency in environment.yaml, but not used.

Future work

Similar progress bars could be added in other places, like scoring and inference.
It would be nice to trim unused dependencies.
The max of 20 workers in the process pool seems arbitrary.

I see there aren't contributing guidelines and this repo is just a code dump. Opening this PR as a way to test the waters - I would be happy to see a community form around this tool. I am curious to see what problems people are "throwing compute" at.

LLM Usage

Code written by @claude and reviewed by me.
Code "tested" by me.

PR message written by me.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant