wta: Add support for --transitions to bench.py

This allows to call the generator not only with --zero-frequency, but also with
--transitions.