(self, encode_config=True)
| 157 | json.dump(self.all_models, f, indent=4, cls=ModelConfigEncoder) |
| 158 | |
| 159 | def read_json(self, encode_config=True): |
| 160 | with open(self.model_file, "r") as f: |
| 161 | out = json.load(f) |
| 162 | |
| 163 | if not encode_config: |
| 164 | return out |
| 165 | |
| 166 | for cc, models in out.items(): |
| 167 | for base_model, configs in models.items(): |
| 168 | for i in range(len(configs)): |
| 169 | out[cc][base_model][i]["config"] = ModelConfig( |
| 170 | **configs[i]["config"] |
| 171 | ) |
| 172 | return out |
| 173 | |
| 174 | def available_models(self): |
| 175 | available = self.all_models.get(self.cc, {}) |
no test coverage detected