-
-
Notifications
You must be signed in to change notification settings - Fork 41
Train related tests #491
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
Train related tests #491
Conversation
tests/training/test_train.py
Outdated
| def wandb_offline_env(monkeypatch, session_tmp_path): | ||
| """Put W&B offline, quiet; force CPU.""" | ||
| save_dir = str(session_tmp_path / "wandb") | ||
| monkeypatch.setenv("WANDB_SILENT", "true") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We should avoid using environmental variables if we can. It adds complexity to the tests. Are these two variables needed for the test or do they clean up the test outputs or something?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@felix-e-h-p thanks for the updates, I'm guessing the env vars weren't needed?
You might want to dropmonkeypatch now that its no longer used in this fixture and maybe update the fixture name and doc to reflect what it does now
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ahh yeah! Sorry - well spotted. Will push now, finalised you reckon?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good! Thanks for doing this!
Relating to #432
Also updated relative to data sampler modifications