Caffin/bert-dllm

1from diffusion import DiffusionConfig2 3 4def test_article_schedule_has_ten_rates_and_resolves_the_canvas() -> None:5    config = DiffusionConfig()6 7    assert config.mask_probabilities == (1.0, 0.9, 0.8, 0.7, 0.6, 0.5, 0.4, 0.3, 0.2, 0.1)8    assert config.masks_after_step(1) == 2169    assert config.masks_after_step(10) == 010