(pipeline_class_name)
| 214 | |
| 215 | def _get_task_class(mapping, pipeline_class_name, throw_error_if_not_exist: bool = True): |
| 216 | def get_model(pipeline_class_name): |
| 217 | for task_mapping in SUPPORTED_TASKS_MAPPINGS: |
| 218 | for model_name, pipeline in task_mapping.items(): |
| 219 | if pipeline.__name__ == pipeline_class_name: |
| 220 | return model_name |
| 221 | |
| 222 | model_name = get_model(pipeline_class_name) |
| 223 |