(pooling_metadata: PoolingMetadata)
| 77 | |
| 78 | |
| 79 | def get_tasks(pooling_metadata: PoolingMetadata) -> list[PoolingTask]: |
| 80 | pooling_params = get_pooling_params(pooling_metadata) |
| 81 | tasks: list[PoolingTask] = [task for pooling_param in pooling_params if (task := pooling_param.task) is not None] |
| 82 | assert len(pooling_params) == len(tasks) |
| 83 | |
| 84 | return tasks |
| 85 | |
| 86 | |
| 87 | def get_prompt_token_ids(pooling_metadata: PoolingMetadata) -> list[paddle.Tensor]: |
no test coverage detected