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

Fix wandb integration with SetFit model #30021

Merged

Conversation

timothepearce
Copy link
Contributor

What does this PR do?

This PR fixes an error popping up while training a SetFitModelwhile reporting to wandb.

To fix this issue, we could also subclass and override the setup method specifically in the setfit repository, but that would be a much bigger job. @tomaarsen if think you're the one who should decide on that.

Fixes # (issue)
huggingface/setfit#498

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you read the contributor guideline,
    Pull Request section?
  • Was this discussed/approved via a Github issue or the forum? Please add a link
    to it if that's the case.
  • Did you make sure to update the documentation with your changes? Here are the
    documentation guidelines, and
    here are tips on formatting docstrings.
  • Did you write any new necessary tests?

Who can review?

@tomaarsen @ArthurZucker

Copy link
Collaborator

@ArthurZucker ArthurZucker left a comment

Choose a reason for hiding this comment

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

lgtm, can you change the status of the PR to ready

@timothepearce timothepearce marked this pull request as ready for review May 1, 2024 15:33
@timothepearce
Copy link
Contributor Author

@ArthurZucker Done!

Copy link
Member

@tomaarsen tomaarsen 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, I appreciate having this fix in the upstream rather than in SetFit itself :)

@ArthurZucker
Copy link
Collaborator

Just need the CIs to go Green 😉 you can rebase on main and run make fixup! 🤗

@timothepearce timothepearce force-pushed the fix-wandb-integration-with-setfit branch from f46c214 to 76ef8f6 Compare May 22, 2024 08:24
@timothepearce
Copy link
Contributor Author

@ArthurZucker That's what I did, but run_testsand ci/circleci: tests_tf are still failing, as it on the main branch. 😄

@ArthurZucker
Copy link
Collaborator

Just re-triggered them!

@timothepearce timothepearce force-pushed the fix-wandb-integration-with-setfit branch from 0b6b42f to 76ef8f6 Compare May 24, 2024 07:55
@timothepearce timothepearce force-pushed the fix-wandb-integration-with-setfit branch from 76ef8f6 to 3abbdb4 Compare May 24, 2024 08:09
@timothepearce timothepearce force-pushed the fix-wandb-integration-with-setfit branch from 3abbdb4 to 107d5df Compare June 12, 2024 12:24
@ArthurZucker
Copy link
Collaborator

the test seems unrelated so I think we can merge!

@ArthurZucker ArthurZucker merged commit 3b5fa14 into huggingface:main Jun 19, 2024
19 of 21 checks passed
@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

itazap pushed a commit that referenced this pull request Jun 20, 2024
Fix W&B integration with SetFit model

Co-authored-by: PEARCE Timothe <[email protected]>
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.

model_config = model.config.to_dict() - AttributeError: 'dict' object has no attribute 'to_dict'
4 participants