8000 Disable report callbacks for certain training tests by ivarflakstad · Pull Request #38088 · huggingface/transformers · GitHub
[go: up one dir, main page]

Skip to content

Disable report callbacks for certain training tests #38088

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

Merged
merged 3 commits into from
May 13, 2025

Conversation

ivarflakstad
Copy link
Member
@ivarflakstad ivarflakstad commented May 12, 2025

This fixes the No module named 'pynvml' error in the affected tests on non-cuda devices.

It does not address test_auto_batch_size_finder however, as that test runs run_glue.py directly.

I tried adding --report_to none as an argument to the run_glue.py call. I'll check if it actually works.

@ivarflakstad ivarflakstad requested a review from ydshieh May 12, 2025 16:49
@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.

Copy link
Collaborator
@ydshieh ydshieh left a comment

Choose a reason for hiding this comment

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

OK, thanks! (as long as the glue one works)

@ivarflakstad ivarflakstad marked this pull request as ready for review May 13, 2025 10:12
@ivarflakstad ivarflakstad merged commit e27d230 into main May 13, 2025
12 checks passed
@ivarflakstad ivarflakstad deleted the disable-report-callbacks-in-training-tests branch May 13, 2025 12:49
zucchini-nlp pushed a commit to zucchini-nlp/transformers that referenced this pull request May 14, 2025
* Disable report callbacks for certain training tests

* Disable report callbacks for test_auto_batch_size_finder
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
0