Unverified Commit 46342a74 authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

Fix pai pipeline (#2123)

parent 475a57d8
...@@ -36,11 +36,11 @@ def update_training_service_config(args): ...@@ -36,11 +36,11 @@ def update_training_service_config(args):
config[args.ts]['paiConfig']['token'] = args.pai_token config[args.ts]['paiConfig']['token'] = args.pai_token
if args.nni_docker_image is not None: if args.nni_docker_image is not None:
config[args.ts]['trial']['image'] = args.nni_docker_image config[args.ts]['trial']['image'] = args.nni_docker_image
if args.nniManagerNFSMountPath is not None: if args.nni_manager_nfs_mount_path is not None:
config[args.ts]['trial']['nniManagerNFSMountPath'] = args.nni_manager_nfs_mount_path config[args.ts]['trial']['nniManagerNFSMountPath'] = args.nni_manager_nfs_mount_path
if args.containerNFSMountPath is not None: if args.container_nfs_mount_path is not None:
config[args.ts]['trial']['containerNFSMountPath'] = args.container_nfs_mount_path config[args.ts]['trial']['containerNFSMountPath'] = args.container_nfs_mount_path
if args.paiStoragePlugin is not None: if args.pai_storage_plugin is not None:
config[args.ts]['trial']['paiStoragePlugin'] = args.pai_storage_plugin config[args.ts]['trial']['paiStoragePlugin'] = args.pai_storage_plugin
if args.vc is not None: if args.vc is not None:
config[args.ts]['trial']['virtualCluster'] = args.vc config[args.ts]['trial']['virtualCluster'] = args.vc
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment