-
Notifications
You must be signed in to change notification settings - Fork 37
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
Quality of life smart validate
Improvements
#458
Conversation
- Set `CUDA_VISIBLE_DEVICES` environment variable within `smart validate` prioro to importing any ML deps to prevent false negatives on multi-GPU systems - Move SmartRedis logs from standard out to dedicated log file in the validation temporary directory - Suppress `sklearn` deprecation warning by pinning `KMeans` constructor argument
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.
LGTM, I love quality of life improvements!
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## develop #458 +/- ##
===========================================
+ Coverage 90.42% 90.80% +0.38%
===========================================
Files 60 60
Lines 3738 3830 +92
===========================================
+ Hits 3380 3478 +98
+ Misses 358 352 -6 |
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 finding this
smartsim/_core/_cli/validate.py
Outdated
if with_pt: | ||
logger.info("Verifying Torch Backend") | ||
_test_torch_install(client, device) | ||
if with_tf: |
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.
If this fixes the problem we found, could you move the Tensorflow test to the very end?
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.
absolutely!!
Quality of life `smart validate` improvements: - Set `CUDA_VISIBLE_DEVICES` environment variable within `smart validate` prior to importing any ML deps to prevent false negatives on multi-GPU systems - Move SmartRedis logs from standard out to dedicated log file in the validation temporary directory - Suppress `sklearn` deprecation warning by pinning `KMeans` constructor argument - Move TF test to last as TF may reserve the GPUs it uses [ committed by @MattToast ] [ reviewed by @al-rigazzi @ashao ]
Quality of life
smart validate
improvements:CUDA_VISIBLE_DEVICES
environment variable withinsmart validate
prior to importing any ML deps to prevent false negatives on multi-GPU systemssklearn
deprecation warning by pinningKMeans
constructor argument