AML WS Connection to ACR

ws_acr_connection.py

import os
from azure.ai.ml import MLClient
from azure.ai.ml.entities import WorkspaceConnection
from azure.identity import DefaultAzureCredential
from azure.ai.ml.entities import UsernamePasswordConfiguration

# Enter details of your AML workspace
subscription_id = ""
resource_group = ""
workspace_name = ""

# get a handle to the workspace
ml_client = MLClient(
    DefaultAzureCredential(), subscription_id, resource_group, workspace_name
)

# verify existing connections
for conn in ml_client.connections.list():
    print(conn)
    print("==============")

###################### if connection exists. QUIT

# fetching secrets from env var to secure access, these secrets can be set outside or source code
acr_username = ""  # fill in e.g. testacr
acr_password = ""

credentials = UsernamePasswordConfiguration(
    username=acr_username, password=acr_password
)

ws_connection = WorkspaceConnection(
    name="", # fill in e.g. testacr.azurecr.io
    target="",  # fill in e.g. testacr.azurecr.io
    type="container_registry",
    credentials=credentials,
)

ml_client.connections.create_or_update(ws_connection)