mlos_bench.services.remote.azure.azure_auth

A collection Service functions for managing VMs on Azure.

Classes

AzureAuthService

Helper methods to get access to Azure services.

Module Contents

class mlos_bench.services.remote.azure.azure_auth.AzureAuthService(config: Dict[str, Any] | None = None, global_config: Dict[str, Any] | None = None, parent: mlos_bench.services.base_service.Service | None = None, methods: Dict[str, Callable] | List[Callable] | None = None)[source]

Bases: mlos_bench.services.base_service.Service, mlos_bench.services.types.authenticator_type.SupportsAuth[azure.core.credentials.TokenCredential]

Helper methods to get access to Azure services.

Create a new instance of Azure authentication services proxy.

Parameters:

  • config (dict) – Free-format dictionary that contains the benchmark environment configuration.

  • global_config (dict) – Free-format dictionary of global parameters.

  • parent (Service) – Parent service that can provide mixin functions.

  • methods (Union[Dict[str, Callable], List[Callable], None]) – New methods to register with the service.

get_access_token() → str[source]

Get the access token from Azure CLI, if expired.

Return type:

str

get_auth_headers() → dict[source]

Get the authorization part of HTTP headers for REST API calls.

Return type:

dict

get_credential() → azure.core.credentials.TokenCredential[source]

Return the Azure SDK credential object.

Return type:

azure.core.credentials.TokenCredential