Update pipeline-train.py
This commit is contained in:
Родитель
fbd6227848
Коммит
cc2eba43e4
|
@ -19,19 +19,19 @@ def get_config_parger(parser: argparse.ArgumentParser = None):
|
||||||
group.add_argument(
|
group.add_argument(
|
||||||
"--subscription_id",
|
"--subscription_id",
|
||||||
type=str,
|
type=str,
|
||||||
required=True,
|
required=False,
|
||||||
help="Subscription ID",
|
help="Subscription ID",
|
||||||
)
|
)
|
||||||
group.add_argument(
|
group.add_argument(
|
||||||
"--resource_group",
|
"--resource_group",
|
||||||
type=str,
|
type=str,
|
||||||
required=True,
|
required=False,
|
||||||
help="Resource group name",
|
help="Resource group name",
|
||||||
)
|
)
|
||||||
group.add_argument(
|
group.add_argument(
|
||||||
"--workspace_name",
|
"--workspace_name",
|
||||||
type=str,
|
type=str,
|
||||||
required=True,
|
required=False,
|
||||||
help="Workspace name",
|
help="Workspace name",
|
||||||
)
|
)
|
||||||
group.add_argument(
|
group.add_argument(
|
||||||
|
@ -129,14 +129,6 @@ def connect_to_aml(args):
|
||||||
print(
|
print(
|
||||||
"Could not find config.json, using config.yaml refs to Azure ML workspace instead."
|
"Could not find config.json, using config.yaml refs to Azure ML workspace instead."
|
||||||
)
|
)
|
||||||
|
|
||||||
# tries to connect using cli args if provided else using config.yaml
|
|
||||||
ml_client = MLClient(
|
|
||||||
subscription_id=args.subscription_id,
|
|
||||||
resource_group_name=args.resource_group,
|
|
||||||
workspace_name=args.workspace_name,
|
|
||||||
credential=credential,
|
|
||||||
)
|
|
||||||
return ml_client
|
return ml_client
|
||||||
|
|
||||||
|
|
||||||
|
@ -396,4 +388,4 @@ def main():
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
Загрузка…
Ссылка в новой задаче