Interesting use case. I’m still unsure what’s causing the dict creation, but I guess that something might have changed in the object such that the collate_fn is now treating it differently.
Could you try to create a custom collate_fn and return the desired type?