[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:
Ameer Haj Ali
2020-07-16 09:54:20 -07:00
committed by GitHub
co-authored by Eric Liang Ameer Haj Ali
parent 63e052a5f3
commit 1e46d4e29f
8 changed files with 47 additions and 31 deletions
@@ -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)