(self, model_name)
| 150 | |
| 151 | |
| 152 | def fetch_model(self, model_name): |
| 153 | fetched_models = [] |
| 154 | for model, model_path, model_name_ in zip(self.model, self.model_path, self.model_name): |
| 155 | if model_name == model_name_: |
| 156 | fetched_models.append(model) |
| 157 | if len(fetched_models) == 0: |
| 158 | print(f"No {model_name} models available.") |
| 159 | return None |
| 160 | if len(fetched_models) > 1: |
| 161 | raise ValueError( |
| 162 | f"Expected one {model_name} model, but found {len(fetched_models)}." |
| 163 | ) |
| 164 | return fetched_models[0] |