From a776cb0de3a64d8fafde02da922ec08c0db91500 Mon Sep 17 00:00:00 2001 From: Anna Pfohl Date: Wed, 29 Nov 2023 10:07:06 -0800 Subject: [PATCH] prompts -> inputs --- scripts/inference/endpoint_generate.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/scripts/inference/endpoint_generate.py b/scripts/inference/endpoint_generate.py index d94676a2e1..e78fecf59b 100644 --- a/scripts/inference/endpoint_generate.py +++ b/scripts/inference/endpoint_generate.py @@ -39,8 +39,8 @@ def parse_args() -> Namespace: ##### # Path Parameters parser.add_argument( - '-p', - '--prompts', + '-i', + '--inputs', nargs='+', help=f'List of strings, local datafiles (starting with {utils.PROMPTFILE_PREFIX}),' +\ ' and/or remote object stores' @@ -114,10 +114,10 @@ async def main(args: Namespace) -> None: if not api_key: log.warning(f'API key not set in {ENDPOINT_API_KEY_ENV}') - new_prompts = [] - for prompt in args.prompts: + new_inputs = [] + for prompt in args.inputs: if prompt.startswith(utils.PROMPTFILE_PREFIX): - new_prompts.append(prompt) + new_inputs.append(prompt) continue input_object_store = maybe_create_object_store_from_uri(prompt) @@ -127,9 +127,9 @@ async def main(args: Namespace) -> None: log.info(f'Downloaded {prompt} to {local_output_path}') prompt = f'{utils.PROMPTFILE_PREFIX}{local_output_path}' - new_prompts.append(prompt) + new_inputs.append(prompt) - prompt_strings = utils.load_prompts(new_prompts, args.prompt_delimiter) + prompt_strings = utils.load_prompts(new_inputs, args.prompt_delimiter) cols = ['batch', 'prompt', 'output'] param_data = {