Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

make jax callback compatible with tqdm #20

Merged
merged 2 commits into from
May 15, 2024
Merged

make jax callback compatible with tqdm #20

merged 2 commits into from
May 15, 2024

Conversation

mdmould
Copy link
Contributor

@mdmould mdmould commented May 14, 2024

This is a fix for #19.

@mdmould
Copy link
Contributor Author

mdmould commented May 14, 2024

This second commit makes fixes following discussion from #12 (comment) onwards.

jax_tqdm/pbar.py Show resolved Hide resolved
jax_tqdm/pbar.py Show resolved Hide resolved
@zombie-einstein
Copy link
Collaborator

@jeremiecoullon happy to merge? Then I can get these fixes into a new release.

@jeremiecoullon
Copy link
Owner

This all looks good! @mdmould thanks for the PR 🙏

@zombie-einstein zombie-einstein merged commit cd2fe02 into jeremiecoullon:main May 15, 2024
3 checks passed
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.

3 participants