Update pipeline-train.py
This commit is contained in:
Родитель
8b3774f1f8
Коммит
64ff3185d2
|
@ -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(
|
||||
|
|
Загрузка…
Ссылка в новой задаче