(model: str | ModelName)
| 801 | |
| 802 | |
| 803 | def _get_model_info(model: str | ModelName) -> ModelInfo | None: |
| 804 | if isinstance(model, str): |
| 805 | return MODEL_INFO.get(model) |
| 806 | return MODEL_INFO.get(model.value) |
| 807 | |
| 808 | |
| 809 | def _normalize_model_names(models: List[str | ModelName]) -> List[str]: |
no test coverage detected
searching dependent graphs…