diff --git a/nlp/python-sdk-v2/mlops/azureml/train/pipeline-train.py b/nlp/python-sdk-v2/mlops/azureml/train/pipeline-train.py
index b980cd2..28198ec 100644
--- a/nlp/python-sdk-v2/mlops/azureml/train/pipeline-train.py
+++ b/nlp/python-sdk-v2/mlops/azureml/train/pipeline-train.py
@@ -8,6 +8,7 @@ from azure.ai.ml import MLClient
 from azure.ai.ml import command
 from azure.ai.ml import Input, Output
 from azure.ai.ml import dsl, Input, Output
+import json
 
 
 def get_config_parger(parser: argparse.ArgumentParser = None):
@@ -16,22 +17,28 @@ def get_config_parger(parser: argparse.ArgumentParser = None):
         parser = argparse.ArgumentParser(description=__doc__)
 
     group = parser.add_argument_group("Azure ML references")
+    group.add_argument(
+        "--config_location",
+        type=str,
+        required=False,
+        help="Subscription ID",
+    )
     group.add_argument(
         "--subscription_id",
         type=str,
-        required=True,
+        required=False,
         help="Subscription ID",
     )
     group.add_argument(
         "--resource_group",
         type=str,
-        required=True,
+        required=False,
         help="Resource group name",
     )
     group.add_argument(
         "--workspace_name",
         type=str,
-        required=True,
+        required=False,
         help="Workspace name",
     )
     group.add_argument(
@@ -123,7 +130,7 @@ def connect_to_aml(args):
     # Get a handle to workspace
     try:
         # ml_client to connect using local config.json
-        ml_client = MLClient.from_config(credential=credential)
+        ml_client = ml_client = MLClient.from_config(credential, path='config.json')
 
     except Exception as ex:
         print(