-
Notifications
You must be signed in to change notification settings - Fork 365
feat: enable reduction server #741
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
feat: enable reduction server #741
Conversation
da33a28
to
99312e6
Compare
reduction_server_machine_type=reduction_server_machine_type, | ||
) | ||
|
||
worker_pool_specs = worker_pool_specs.pool_specs |
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.
This changes the type of the variable. Although allowable in Python, prefer not to do that for readability purposes.
@@ -86,6 +91,10 @@ def spec_dict(self) -> Dict[str, Union[int, str, Dict[str, Union[int, str]]]]: | |||
spec["machine_spec"]["accelerator_type"] = accelerator_type | |||
spec["machine_spec"]["accelerator_count"] = self.accelerator_count | |||
|
|||
if self.container_uri == REDUCTION_SERVER_CONTAINER_URI: |
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.
For now, I think it may be more extendable to add container_spec
whenever self.container_uri
is populated.
99312e6
to
a51daae
Compare
7fe79b0
to
d473d24
Compare
d473d24
to
9dac2a9
Compare
Fixes #<b/197889114> 🦕