We have custom method self.machineType in work items class which essentially will assign machine types based on the request module to a VM. The machine type information could be passed in the module instructions which would allow removing the customization from the work item.