Add some protections against accessing PartitioningScheme non-iterable structure