def set_choice_set_size(self): if self.sampler_size is None: self.sampler_size = 0 for nest, values in self.nested_structure.iteritems(): self.sampler_size += len(values) ChoiceModel.set_choice_set_size(self)