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 prediction async #54

Merged
merged 7 commits into from
Dec 17, 2024
Merged

Make prediction async #54

merged 7 commits into from
Dec 17, 2024

Conversation

davidotte
Copy link
Collaborator

@davidotte davidotte commented Dec 13, 2024

Change Description

Try to be precise. You can additionally add comments to your PR, this might help the reviewer a lot.

This PR involves the client side for making predictions async. Please also check the corresponding Server PR. On the client side, we only need updates in client.py. We now send streaming requests and basically just loop over all messages we receive. Currently, if a duration estimation is received, I print the estimated duration and then show a progress bar that runs for this estimated duration. The question is now if we want to keep this initial print of estimated duration and if we might also want to add a message that is printed if the prediction is taking longer than expected.

If you used new dependencies: Did you add them to requirements.txt?

Yes. Should I also add version numbers?

Who did you ping on Mattermost to review your PR? Please ping that person again whenever you are ready for another review.

@noahho

Breaking changes

If you made any breaking changes, please update the version number.
Breaking changes are totally fine, we just need to make sure to keep the users informed and the server in sync.

Does this PR break the API? If so, what is the corresponding server commit?

Yes, PR mentioned above

Does this PR break the user interface? If so, why?


Please do not mark comments/conversations as resolved unless you are the assigned reviewer. This helps maintain clarity during the review process.

@davidotte davidotte requested a review from noahho December 13, 2024 16:15
tabpfn_client/client.py Outdated Show resolved Hide resolved
Copy link
Collaborator

@noahho noahho left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me

Copy link
Collaborator

@SamuelGabriel SamuelGabriel left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this pr looks fine to me :)

@davidotte davidotte force-pushed the make-prediction-async branch from e7e27c8 to 8e65d35 Compare December 17, 2024 13:57
@davidotte davidotte merged commit 71bfa89 into main Dec 17, 2024
2 checks passed
@davidotte davidotte deleted the make-prediction-async branch December 17, 2024 13:58
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