mirror of
https://github.com/wassname/ray.git
synced 2026-09-10 12:38:43 +08:00
[Autoscaler] Making bootstrap config part of the node provider interface (#9443)
* supporting custom bootstrap config for external node providers * bootstrap config * renamed config to cluster_config * lint * remove 2 args from importer * complete move of bootstrap to node_provider * renamed provider_cls * move imports outside functions * lint * Update python/ray/autoscaler/node_provider.py Co-authored-by: Eric Liang <ekhliang@gmail.com> * final fixes * keeping lines to reduce diff * lint * lamba config * filling in -> adding for lint Co-authored-by: Ameer Haj Ali <ameerhajali@Ameers-MacBook-Pro.local> Co-authored-by: Eric Liang <ekhliang@gmail.com>
This commit is contained in:
co-authored by
Eric Liang
Ameer Haj Ali
parent
63e052a5f3
commit
1e46d4e29f
@@ -2,6 +2,7 @@ import logging
|
||||
|
||||
from ray.autoscaler.kubernetes import core_api, log_prefix
|
||||
from ray.autoscaler.node_provider import NodeProvider
|
||||
from ray.autoscaler.kubernetes.config import bootstrap_kubernetes
|
||||
from ray.autoscaler.tags import TAG_RAY_CLUSTER_NAME
|
||||
from ray.autoscaler.updater import KubernetesCommandRunner
|
||||
|
||||
@@ -97,3 +98,7 @@ class KubernetesNodeProvider(NodeProvider):
|
||||
docker_config=None):
|
||||
return KubernetesCommandRunner(log_prefix, self.namespace, node_id,
|
||||
auth_config, process_runner)
|
||||
|
||||
@staticmethod
|
||||
def bootstrap_config(cluster_config):
|
||||
return bootstrap_kubernetes(cluster_config)
|
||||
|
||||
Reference in New Issue
Block a user