Pass `num_warmup` as kwarg to `step_warmup` - #178

Merged

penelopeysm merged 2 commits into

mainfrom

py/pass-num-warmup

Oct 7, 2025

Commits on Oct 7, 2025

Pass `num_warmup` kwarg to `step_warmup()`
penelopeysm
committed
6d70f71
Mention the num_warmup kwarg in step_warmup's docstring
penelopeysm
committed
fdab17c