Merge pull request microsoft#51 from microsoft/eedorenko/auth-error-fix

Authentication error fix

Author: eedorenko

docs/getting_started.md

@@ -127,7 +127,7 @@ Add a command line step **Run Training Pipeline** with the following script:
 ```bash
 docker run  -v $(System.DefaultWorkingDirectory)/_ci-build/mlops-pipelines/ml_service/pipelines:/pipelines \
 -w=/pipelines -e MODEL_NAME=$MODEL_NAME -e EXPERIMENT_NAME=$EXPERIMENT_NAME \
--e TENANT_ID=$TENANT_ID -e SP_APP_ID=$SP_APP_ID -e SP_APP_SECRET=$SP_APP_SECRET \
+-e TENANT_ID=$TENANT_ID -e SP_APP_ID=$SP_APP_ID -e SP_APP_SECRET=$(SP_APP_SECRET) \
 mcr.microsoft.com/mlops/python:latest python run_train_pipeline.py
 ```
 

environment_setup/docker-image-pipeline.yml

@@ -15,7 +15,7 @@ trigger:
 
 variables:
   containerRegistry: $[coalesce(variables['acrServiceConnection'], 'acrconnection')]
-  imageName: $[coalesce(variables['agentImageName'], 'public/mlops/mlopspython')]  
+  imageName: $[coalesce(variables['agentImageName'], 'public/mlops/python')]  
 
 steps:
   - task: Docker@2

ml_service/pipelines/run_train_pipeline.py

@@ -2,9 +2,13 @@
 import os
 import json
 import requests
-from azureml.core.authentication import AzureCliAuthentication
+from azure.common.credentials import ServicePrincipalCredentials
 
 
+tenant_id = os.environ.get("TENANT_ID")
+app_id = os.environ.get("SP_APP_ID")
+app_secret = os.environ.get("SP_APP_SECRET")
+
 try:
     with open("train_pipeline.json") as f:
         train_pipeline_json = json.load(f)
@@ -15,13 +19,20 @@
 experiment_name = os.environ.get("EXPERIMENT_NAME")
 model_name = os.environ.get("MODEL_NAME")
 
-cli_auth = AzureCliAuthentication()
-token = cli_auth.get_authentication_header()
+credentials = ServicePrincipalCredentials(
+    client_id=app_id,
+    secret=app_secret,
+    tenant=tenant_id
+)
+
+token = credentials.token['access_token']
+print("token", token)
+auth_header = {"Authorization": "Bearer " + token}
 
 rest_endpoint = train_pipeline_json["rest_endpoint"]
 
 response = requests.post(
-    rest_endpoint, headers=token,
+    rest_endpoint, headers=auth_header,
     json={"ExperimentName": experiment_name,
           "ParameterAssignments": {"model_name": model_name}}
 )