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

Colocated model fix #237

Merged
merged 3 commits into from
Oct 5, 2022
Merged

Conversation

mellis13
Copy link
Contributor

@mellis13 mellis13 commented Oct 4, 2022

A bug was reported that uninitialized variables in the colocated.py entry point were causing an error when utilizing add_ml_model() for PyTorch models. This PR fixes that bug and adds tests for the add_ml_model() methods using PyTorch.

Also added tests using PyTorch to cover the cases that led to the
error.
@codecov-commenter
Copy link

codecov-commenter commented Oct 4, 2022

Codecov Report

Merging #237 (a198058) into develop (c1dedc4) will increase coverage by 0.29%.
The diff coverage is n/a.

Additional details and impacted files

Impacted file tree graph

@@             Coverage Diff             @@
##           develop     #237      +/-   ##
===========================================
+ Coverage    84.14%   84.43%   +0.29%     
===========================================
  Files           59       56       -3     
  Lines         3317     3284      -33     
===========================================
- Hits          2791     2773      -18     
+ Misses         526      511      -15     
Impacted Files Coverage Δ
smartsim/__init__.py
smartsim/tf/__init__.py
smartsim/tf/utils.py
smartsim/_core/launcher/taskManager.py 82.24% <0.00%> (+2.36%) ⬆️

catch error when PyTorch is not installed.
Copy link
Member

@MattToast MattToast left a comment

Choose a reason for hiding this comment

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

LGTM!! Thanks for taking care of this!

@mellis13 mellis13 merged commit be6dfc3 into CrayLabs:develop Oct 5, 2022
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