mstar.model.cosmos3.tests.stubs# Stand-ins the Cosmos3 DiT tests share. Classes StubSampler(timesteps[, update]) A request's sampler reduced to its schedule: update is the whole step and nothing rides the loop-back edges. class mstar.model.cosmos3.tests.stubs.StubSampler(timesteps, update=lambda sample, velocity: ...)[source]# Bases: object A request’s sampler reduced to its schedule: update is the whole step and nothing rides the loop-back edges. Pair it with sub._solver_state = (). initial_state(sample)[source]# step(model_output, sample, step_index, solver, generator=None)[source]#