Skip to content

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.

4 participants