| from diffusion import DiffusionConfig | |
| def test_article_schedule_has_ten_rates_and_resolves_the_canvas() -> None: | |
| config = DiffusionConfig() | |
| assert config.mask_probabilities == (1.0, 0.9, 0.8, 0.7, 0.6, 0.5, 0.4, 0.3, 0.2, 0.1) | |
| assert config.masks_after_step(1) == 216 | |
| assert config.masks_after_step(10) == 0 | |