Add lambda in calls to execute_with_retry (#2733)

Co-authored-by: narrieta <narrieta>
This commit is contained in:
Norberto Arrieta 2023-01-19 17:03:25 -08:00 коммит произвёл GitHub
Родитель 4b8fe5ebda
Коммит 43b7c3c671
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 3 добавлений и 3 удалений

Просмотреть файл

@ -86,7 +86,7 @@ class VirtualMachineBaseClass(ABC):
class VirtualMachine(VirtualMachineBaseClass):
def get_instance_view(self) -> VirtualMachineInstanceView:
log.info("Retrieving instance view for %s", self._identifier)
return execute_with_retry(self._compute_client.virtual_machines.get(
return execute_with_retry(lambda: self._compute_client.virtual_machines.get(
resource_group_name=self._identifier.resource_group,
vm_name=self._identifier.name,
expand="instanceView"
@ -94,7 +94,7 @@ class VirtualMachine(VirtualMachineBaseClass):
def get_extensions(self) -> List[VirtualMachineExtension]:
log.info("Retrieving extensions for %s", self._identifier)
return execute_with_retry(self._compute_client.virtual_machine_extensions.list(
return execute_with_retry(lambda: self._compute_client.virtual_machine_extensions.list(
resource_group_name=self._identifier.resource_group,
vm_name=self._identifier.name))
@ -133,7 +133,7 @@ class VmScaleSet(VirtualMachineBaseClass):
def get_extensions(self) -> List[VirtualMachineScaleSetExtension]:
log.info("Retrieving extensions for %s", self._identifier)
return execute_with_retry(self._compute_client.virtual_machine_scale_set_extensions.list(
return execute_with_retry(lambda: self._compute_client.virtual_machine_scale_set_extensions.list(
resource_group_name=self._identifier.resource_group,
vm_scale_set_name=self._identifier.name))