-
Notifications
You must be signed in to change notification settings - Fork 1.2k
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
Updated schedule logic for placement groups for ray backend #2523
Conversation
for more information, see https://pre-commit.ci
…ludwig into updated-schedule-logic-for-pg
@magdyksaleh I'm assuming you want this to be included in 0.6? In that case, you should add the |
Also, you should add a test to make sure this works as expected. |
for more information, see https://pre-commit.ci
Added |
@@ -812,17 +812,24 @@ def initialize(self): | |||
# Disable placement groups on dask | |||
dask.config.set(annotations={"ray_remote_args": {"placement_group": None}}) | |||
|
|||
def generate_bundles(self, num_cpu): |
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.
Nit: prefix with underscore since this is not really meant to be used outside this package.
No description provided.