Skip to content

Conversation

KPostOffice
Copy link
Contributor

…y configurable

I also added a note to each schedulers init method to help with maintainablility

I changed the create scheduler function for the local scheduler to specify the key word args that it takes rather than getting them from the **kwargs dict.

Test plan:

  • add unit tests

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 15, 2023
…y configurable

I also added a note to each schedulers __init__ method to help with maintainablility

Signed-off-by: Kevin <kpostlet@redhat.com>
@codecov
Copy link

codecov bot commented Sep 15, 2023

Codecov Report

Merging #767 (509ced3) into main (1f5eac8) will increase coverage by 0.00%.
Report is 3 commits behind head on main.
The diff coverage is 100.00%.

@@           Coverage Diff           @@
##             main     #767   +/-   ##
=======================================
  Coverage   92.88%   92.88%           
=======================================
  Files          96       96           
  Lines        6096     6098    +2     
=======================================
+ Hits         5662     5664    +2     
  Misses        434      434           
Flag Coverage Δ
unittests 92.88% <100.00%> (+<0.01%) ⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Files Coverage Δ
torchx/schedulers/aws_batch_scheduler.py 90.31% <100.00%> (ø)
torchx/schedulers/docker_scheduler.py 94.47% <ø> (ø)
torchx/schedulers/gcp_batch_scheduler.py 91.62% <100.00%> (ø)
torchx/schedulers/kubernetes_mcad_scheduler.py 94.79% <100.00%> (ø)
torchx/schedulers/kubernetes_scheduler.py 92.14% <100.00%> (ø)
torchx/schedulers/local_scheduler.py 94.22% <100.00%> (ø)
torchx/schedulers/lsf_scheduler.py 99.61% <ø> (ø)
torchx/schedulers/ray_scheduler.py 95.55% <100.00%> (ø)
torchx/schedulers/slurm_scheduler.py 97.00% <ø> (ø)

... and 1 file with indirect coverage changes

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

Copy link
Member

@d4l3k d4l3k 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! A unit test would be nice but not critical and fixing lint

)

return RayScheduler(session_name=session_name)
return RayScheduler(session_name=session_name, ray_client=ray_client)
Copy link
Member

Choose a reason for hiding this comment

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

would be nice to add some unit tests for this

Copy link
Contributor Author

Choose a reason for hiding this comment

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

I'll add some unit tests this week 👍🏻

Signed-off-by: Kevin <kpostlet@redhat.com>
@KPostOffice
Copy link
Contributor Author

@d4l3k I've added those unit tests. Feel free to merge if you think the PR is good to go

Copy link
Member

@d4l3k d4l3k 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 adding!

@d4l3k d4l3k merged commit d713116 into pytorch:main Oct 3, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants