It's worth considering that a number of these are forbidden keys for the attributes dict as they are handled elsewhere. Those you mention that are permitted are: ```sample_dimension```, ```instance_dimension```. I'll add these here.