(self)
| 24 | } |
| 25 | |
| 26 | def test_merge_thr_in_cb_model(self): |
| 27 | kwargs = self.base_kwargs.copy() |
| 28 | kwargs["spk_model"] = "cam++" |
| 29 | merge_thr = 0.5 |
| 30 | kwargs["spk_kwargs"] = {"cb_kwargs": {"merge_thr": merge_thr}} |
| 31 | model = AutoModel(**kwargs) |
| 32 | self.assertEqual(model.cb_model.model_config['merge_thr'], merge_thr) |
| 33 | # res = model.generate(input="/test.wav", |
| 34 | # batch_size_s=300) |
| 35 | |
| 36 | def test_progress_callback_called(self): |
| 37 | class DummyModel: |