(self, log_level=None)
| 51 | return config |
| 52 | |
| 53 | def _init(self, log_level=None): |
| 54 | model = SimpleModel(hidden_dim=8) |
| 55 | engine, _, _, _ = deepspeed.initialize(config=self._config(log_level), |
| 56 | model=model, |
| 57 | model_parameters=model.parameters()) |
| 58 | return engine |
| 59 | |
| 60 | def test_init_with_error_log_level(self): |
| 61 | saved = ds_logging.logger.level |
no test coverage detected