diff --git a/src/xpk/core/pathways.py b/src/xpk/core/pathways.py index 71ed27ea..c17b4c8f 100644 --- a/src/xpk/core/pathways.py +++ b/src/xpk/core/pathways.py @@ -60,6 +60,8 @@ def get_pathways_proxy_args(args) -> str: yaml = """- --server_port=29000 - --resource_manager_address={rm_address} - --gcs_scratch_location={args.pathways_gcs_location}""" + if args.additional_pw_proxy_args: + yaml += args.additional_pw_proxy_args.replace(' ', '\n' + ' '*14 +'- ') if args.use_pathways: return yaml.format(args=args, rm_address=get_rm_address(args)) diff --git a/src/xpk/parser/workload.py b/src/xpk/parser/workload.py index fecebef9..adf1d189 100644 --- a/src/xpk/parser/workload.py +++ b/src/xpk/parser/workload.py @@ -244,6 +244,16 @@ def set_workload_parsers(workload_parser): required=False, ) + workload_pathways_workload_arguments.add_argument( + '--additional_pw_proxy_args', + type=str, + default='', + help=( + 'Additional arguments to pass to the proxy server.' + ), + required=False, + ) + add_shared_workload_create_required_arguments([ workload_create_parser_required_arguments, workload_create_pathways_parser_required_arguments,