def get_auth(env_path: str) -> AbstractAuthentication:
"""
:param env_path:
:return:
"""
logger = logging.getLogger(__name__)
if get_key(env_path, "password") != "YOUR_SERVICE_PRINCIPAL_PASSWORD":
logger.debug("Trying to create Workspace with Service Principal")
aml_sp_password = get_key(env_path, "password")
aml_sp_tennant_id = get_key(env_path, "tenant_id")
aml_sp_username = get_key(env_path, "username")
auth = ServicePrincipalAuthentication(
tenant_id=aml_sp_tennant_id,
service_principal_id=aml_sp_username,
service_principal_password=aml_sp_password,
)
else:
logger.debug("Trying to create Workspace with CLI Authentication")
try:
auth = AzureCliAuthentication()
auth.get_authentication_header()
except AuthenticationException:
logger.debug("Trying to create Workspace with Interactive login")
auth = InteractiveLoginAuthentication()
return auth
def _get_auth():
"""Returns authentication to Azure Machine Learning workspace."""
logger = logging.getLogger(__name__)
if os.environ.get("AML_SP_PASSWORD", None):
logger.debug("Trying to authenticate with Service Principal")
aml_sp_password = os.environ.get("AML_SP_PASSWORD")
aml_sp_tennant_id = os.environ.get("AML_SP_TENNANT_ID")
aml_sp_username = os.environ.get("AML_SP_USERNAME")
auth = ServicePrincipalAuthentication(aml_sp_tennant_id, aml_sp_username, aml_sp_password)
else:
logger.debug("Trying to authenticate with CLI Authentication")
try:
auth = AzureCliAuthentication()
auth.get_authentication_header()
except AuthenticationException:
logger.debug("Trying to authenticate with Interactive login")
auth = InteractiveLoginAuthentication()
return auth
def get_auth(env_path):
logger = logging.getLogger(__name__)
if os.environ.get("AML_SP_PASSWORD", None):
logger.debug("Trying to create Workspace with Service Principal")
aml_sp_password = os.environ.get("AML_SP_PASSWORD")
aml_sp_tennant_id = os.environ.get("AML_SP_TENNANT_ID")
aml_sp_username = os.environ.get("AML_SP_USERNAME")
auth = ServicePrincipalAuthentication(tenant_id=aml_sp_tennant_id,
username=aml_sp_username,
password=aml_sp_password)
else:
logger.debug("Trying to create Workspace with CLI Authentication")
try:
auth = AzureCliAuthentication()
auth.get_authentication_header()
except AuthenticationException:
logger.debug("Trying to create Workspace with Interactive login")
auth = InteractiveLoginAuthentication()
return auth
def get_auth():
"""Get an auth object for use with Workspace objects."""
if os.environ.get("AML_SP_PASSWORD", None):
print("Trying to create Workspace with Service Principal")
aml_sp_password = os.environ.get("AML_SP_PASSWORD")
aml_sp_tenant_id = os.environ.get("AML_SP_TENANT_ID")
aml_sp_username = os.environ.get("AML_SP_USERNAME")
auth = ServicePrincipalAuthentication(
tenant_id=aml_sp_tenant_id,
service_principal_id=aml_sp_username,
service_principal_password=aml_sp_password)
else:
print("Trying to create Workspace with CLI Authentication")
try:
auth = AzureCliAuthentication()
auth.get_authentication_header()
except AuthenticationException:
print("Trying to create Workspace with Interactive login")
auth = InteractiveLoginAuthentication()
return auth
def get_auth(env_path):
logger = logging.getLogger(__name__)
if get_key(env_path, 'password') != "YOUR_SERVICE_PRINCIPAL_PASSWORD":
logger.debug("Trying to create Workspace with Service Principal")
aml_sp_password = get_key(env_path, 'password')
aml_sp_tennant_id = get_key(env_path, 'tenant_id')
aml_sp_username = get_key(env_path, 'username')
auth = ServicePrincipalAuthentication(
tenant_id=aml_sp_tennant_id,
username=aml_sp_username,
password=aml_sp_password
)
else:
logger.debug("Trying to create Workspace with CLI Authentication")
try:
auth = AzureCliAuthentication()
auth.get_authentication_header()
except AuthenticationException:
logger.debug("Trying to create Workspace with Interactive login")
auth = InteractiveLoginAuthentication()
return auth
def get_auth():
logger = logging.getLogger(__name__)
if os.environ.get("AML_SP_PASSWORD", None):
logger.debug("Trying to create Workspace with Service Principal")
aml_sp_password = os.environ.get("AML_SP_PASSWORD")
aml_sp_tennant_id = os.environ.get("AML_SP_TENNANT_ID")
aml_sp_username = os.environ.get("AML_SP_USERNAME")
auth = ServicePrincipalAuthentication(
tenant_id=aml_sp_tennant_id,
username=aml_sp_username,
password=aml_sp_password,
)
else:
logger.debug("Trying to create Workspace with CLI Authentication")
try:
auth = AzureCliAuthentication()
auth.get_authentication_header()
except AuthenticationException:
logger.debug("Trying to create Workspace with Interactive login")
auth = InteractiveLoginAuthentication()
return auth
time.sleep(5)
print("Run status: %s" % run.get_status())
config_json = 'config.json'
with open(config_json, 'r') as f:
config = json.load(f)
try:
svc_pr = ServicePrincipalAuthentication(
tenant_id=config['tenant_id'],
service_principal_id=config['service_principal_id'],
service_principal_password=config['service_principal_password'])
aad_token = svc_pr.get_authentication_header()
except KeyError as e:
print(
"WARNING: No Service Principal found in config.json. This is fine if we are operating in DevOps."
)
svc_pr = None
aad_token = None
pass
ws = Workspace.from_config(path=config_json)
pipeline_name = 'kd_train_the_teacher'
published_pipeline = find_pipeline(ws, pipeline_name)
run_pipeline(ws, published_pipeline, aad_token)