azure_batchai_management_client_list_workspaces_quickstart.py

1import os
2from azure.identity import DefaultAzureCredential
3from azure.mgmt.batchai import BatchAIManagementClient
4
5def run_example():
6    # Substitution of your subscription ID is required
7    subscription_id = os.environ.get("AZURE_SUBSCRIPTION_ID", "your-subscription-id")
8
9    # Authenticate using default Azure credentials (Environment variables, Managed Identity, etc.)
10    credential = DefaultAzureCredential()
11
12    # Initialize the BatchAI Management Client
13    batchai_client = BatchAIManagementClient(
14        credential=credential,
15        subscription_id=subscription_id
16    )
17
18    print("Listing Batch AI Workspaces in the subscription...")
19
20    # List all Batch AI workspaces
21    # Note: As the service is retired, this may return an empty list or an error depending on account status
22    for workspace in batchai_client.workspaces.list():
23        print(f"Workspace Name: {workspace.name}")
24        print(f"Location: {workspace.location}")
25        print("-" * 20)
26
27if __name__ == "__main__":
28    run_example()