I used to be able to get nice grouping with the extra_tasks argument on a DbtTaskGroup but with cosmos I can't replicate it