Moved hyperparameter sampling from state to direct_assignment_representation