Update benchmark.py for simpler Worker API