Looks like you should be able to use regular python variables to do that or eg call a function that returns the right list of workers depending on the environment type.

If the obvious way is what you tried then yeah, the library should be changed to support this.