in python/graphscope/config.py [0:0]
def set_option(self, key, value): # noqa: C901
"""Forward set_option target to actual config fields"""
if key == "addr":
self.coordinator.endpoint = value
elif key == "mode":
self.session.execution_mode = value
elif key == "cluster_type":
self.launcher_type = value
elif key == "k8s_namespace":
self.kubernetes_launcher.namespace = value
elif key == "k8s_image_registry":
self.kubernetes_launcher.image.registry = value
elif key == "k8s_image_repository":
self.kubernetes_launcher.image.repository = value
elif key == "k8s_image_tag":
self.kubernetes_launcher.image.tag = value
elif key == "k8s_image_pull_policy":
self.kubernetes_launcher.image.pull_policy = value
elif key == "k8s_image_secrets":
self.kubernetes_launcher.image.pull_secrets = value
elif key == "k8s_coordinator_cpu":
self.coordinator.resource.set_cpu_request(value)
elif key == "k8s_coordinator_mem":
self.coordinator.resource.set_mem_request(value)
elif key == "etcd_addrs":
self.hosts_launcher.etcd.endpoint = value
elif key == "etcd_listening_client_port":
self.hosts_launcher.etcd.listening_client_port = value
elif key == "etcd_listening_peer_port":
self.hosts_launcher.etcd.listening_peer_port = value
elif key == "k8s_vineyard_image":
self.vineyard.image = value
elif key == "k8s_vineyard_deployment":
self.vineyard.deployment_name = value
elif key == "k8s_vineyard_cpu":
self.vineyard.resource.set_cpu_request(value)
elif key == "k8s_vineyard_mem":
self.vineyard.resource.set_mem_request(value)
elif key == "k8s_engine_cpu":
self.kubernetes_launcher.engine.gae_resource.set_cpu_request(value)
elif key == "k8s_engine_mem":
self.kubernetes_launcher.engine.gae_resource.set_mem_request(value)
elif key == "mars_worker_cpu":
self.kubernetes_launcher.mars.worker_resource.set_cpu_request(value)
elif key == "mars_worker_mem":
self.kubernetes_launcher.mars.worker_resource.set_mem_request(value)
elif key == "mars_scheduler_cpu":
self.kubernetes_launcher.mars.scheduler_resource.set_cpu_request(value)
elif key == "mars_scheduler_mem":
self.kubernetes_launcher.mars.scheduler_resource.set_mem_request(value)
elif key == "k8s_coordinator_pod_node_selector":
self.coordinator.node_selector = base64_encode(json.dumps(value))
elif key == "k8s_engine_pod_node_selector":
self.kubernetes_launcher.engine.node_selector = base64_encode(
json.dumps(value)
)
elif key == "enabled_engines":
self.kubernetes_launcher.engine.enabled_engines = value
elif key == "with_mars":
self.kubernetes_launcher.mars.enable = value
elif key == "with_dataset":
self.kubernetes_launcher.dataset.enable = value
elif key == "k8s_volumes":
self.kubernetes_launcher.volumes = base64_encode(json.dumps(value))
elif key == "k8s_service_type":
self.kubernetes_launcher.service_type = value
elif key == "preemptive":
self.kubernetes_launcher.engine.preemptive = value
elif key == "k8s_deploy_mode":
self.kubernetes_launcher.deployment_mode = value
elif key == "k8s_waiting_for_delete":
self.kubernetes_launcher.waiting_for_delete = value
elif key == "num_workers":
self.session.num_workers = value
self.session.default_local_num_workers = value
elif key == "show_log":
self.show_log = value
elif key == "log_level":
self.log_level = value
elif key == "timeout_seconds":
self.session.timeout_seconds = value
elif key == "dangling_timeout_seconds":
self.session.dangling_timeout_seconds = value
elif key == "dataset_download_retries":
self.hosts_launcher.dataset_download_retries = value
elif key == "k8s_client_config":
self.kubernetes_launcher.config_file = value
elif key == "reconnect":
self.session.reconnect = value
elif key == "vineyard_shared_mem":
pass
else:
raise ValueError("Key not recognized: " + key)