(batch_prompt,demonstrations, task)
| 95 | |
| 96 | |
| 97 | def collate_function(batch_prompt,demonstrations, task): |
| 98 | processed_prompt = [] |
| 99 | for prompt in batch_prompt: |
| 100 | |
| 101 | task_prompt = TASK_PROMT[task] |
| 102 | prompt = prompt[len(task_prompt):] |
| 103 | demonstrations_prompt = "" |
| 104 | for i in range(len(demonstrations["prompt"])): |
| 105 | demonstrations_prompt += demonstrations["prompt"][i] |
| 106 | demonstrations_prompt += demonstrations["answer"][i] |
| 107 | demonstrations_prompt += "\n\n" |
| 108 | |
| 109 | prompt = task_prompt +demonstrations_prompt + Constrained_PROMPT + prompt |
| 110 | processed_prompt.append(prompt) |
| 111 | return processed_prompt |
| 112 | |
| 113 | class API: |
| 114 | def __init__(self, model_path): |
nothing calls this directly
no outgoing calls
no test coverage detected