From 857408874c06ce43131f9748936ba19d5c840341 Mon Sep 17 00:00:00 2001 From: Ian Rodney Date: Fri, 19 Mar 2021 08:39:46 -0700 Subject: [PATCH] [Autoscaler][Azure] check if 'update' is available in (#14787) --- .../autoscaler/_private/azure/node_provider.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/python/ray/autoscaler/_private/azure/node_provider.py b/python/ray/autoscaler/_private/azure/node_provider.py index 2b8aae556..fd594451b 100644 --- a/python/ray/autoscaler/_private/azure/node_provider.py +++ b/python/ray/autoscaler/_private/azure/node_provider.py @@ -223,10 +223,17 @@ class AzureNodeProvider(NodeProvider): """Sets the tag values (string dict) for the specified node.""" node_tags = self._get_cached_node(node_id)["tags"] node_tags.update(tags) - self.compute_client.virtual_machines.update( - resource_group_name=self.provider_config["resource_group"], - vm_name=node_id, - parameters={"tags": node_tags}) + if hasattr(self.compute_client.virtual_machines, "update"): + self.compute_client.virtual_machines.update( + resource_group_name=self.provider_config["resource_group"], + vm_name=node_id, + parameters={"tags": node_tags}) + else: + # Newer versions of the client use begin_update, not update + self.compute_client.virtual_machines.begin_update( + resource_group_name=self.provider_config["resource_group"], + vm_name=node_id, + parameters={"tags": node_tags}) self.cached_nodes[node_id]["tags"] = node_tags def terminate_node(self, node_id):